【字符串, 数学】密钥格式化

题目

482. 密钥格式化

---

给定一个许可密钥字符串 s，仅由字母、数字字符和破折号组成。字符串由 n 个破折号分成 n + 1 组。你也会得到一个整数 k 。

我们想要重新格式化字符串 s，使每一组包含 k 个字符，除了第一组，它可以比 k 短，但仍然必须包含至少一个字符。此外，两组之间必须插入破折号，并且应该将所有小写字母转换为大写字母。

返回 重新格式化的许可密钥 。

示例 1：

输入：S = "5F3Z-2e-9-w", k = 4
输出："5F3Z-2E9W"
解释：字符串 S 被分成了两个部分，每部分 4 个字符；
     注意，两个额外的破折号需要删掉。

示例 2：

输入：S = "2-5g-3-J", k = 2
输出："2-5G-3J"
解释：字符串 S 被分成了 3 个部分，按照前面的规则描述，第一部分的字符可以少于给定的数量，其余部分皆为 2 个字符。

提示:

• 1 <= s.length <= 10^5
• s 只包含字母、数字和破折号 '-'.
• 1 <= k <= 10^4

解题

方法一：字符串 数学

思路

根据题意模拟, 把字符串按 '-' 分割后再加入 StringBuilder 达到去除原本分组的目的。

然后从后往前遍历 sb 并从 0 开始计数(count), 当计数是 k 的倍数时插入 '-'。

注意: 如果遍历到第一个字符且计数正好为 k 的倍数时不用插入。

代码

class Solution {
    public String licenseKeyFormatting(String s, int k) {
        StringBuilder sb = new StringBuilder();
        for (String str : s.split("-")) sb.append(str.toUpperCase());
        int count = 0;
        for (int i = sb.length() - 1; i >= 0; i--) {
            if (++count % k == 0 && i != 0) sb.insert(i, '-');
        }
        return sb.toString();
    }
}