持续创作，加速成长！这是我参与「掘金日新计划 · 10 月更文挑战」的第16天，点击查看活动详情

题目

给定一个许可密钥字符串 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 个字符。

方法一：数学

思路及解法

首先我们计算出字符串 $s$ 中破折号字符的数量 $dashes$，然后通过字符串长度减去 $dashes$ 得到所有非破折号字符的数量为 $chars$，题目要求对取出的字符进行重新分组，使得每个分组恰好包含 $k$ 个字符，且必须满足第一个分组包含的字符个数必须小于等于 $k$，但至少要包含 $1$ 个字符。只需满足第一个分组包含的字符数目刚好等于 $chars \bmod k$，剩余的分组包含的字符数目刚好等于 $k$。

• 首先我们定义一个变量 $i$，初始值为 $0$，我们从头遍历字符串，每次取出字符时首先判断该字符是否为破折号，如果为破折号则跳过；否则将当前的字符计数 $i$ 加 $1$，同时检查如果当前字符为小写字母则将其转化为大写字母，将当前字符加入到字符串 $\textit{ans}$ 的末尾。
• 取出 $chars \bmod k$ 的结果 $mod$。
• 对字符进行计数时，从 $mod$ 开始，每隔 $k$ 个字符就在字符串 $\textit{ans}$ 中添加一个破折号，遍历结束， $\textit{ans}$ 即为返回结果。

代码

class Solution {
    func licenseKeyFormatting(_ s: String, _ k: Int) -> String {
        let dashes = s.filter { $0 == "-" }.count
        let chars = (s.count - dashes)
        let mod: Int = chars % k
        var ans: String = ""
        var i: Int = 0
        for ch: Character in s {
            if ch != "-" {
                if (i - mod) % k == 0 && i > 0 {
                    ans.append("-")
                }
                ans.append(ch.uppercased())
                i += 1
            }
        }
        return ans
    }
}

复杂度分析

• 时间复杂度：$O(N)$，其中 $N$ 为字符串的长度。需要对字符串进行一次遍历。

• 空间复杂度：$O(N)$，其中 $N$ 为字符串的长度。