携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第13天，点击查看活动详情
一、题目描述：
给定一个许可密钥字符串 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
二、思路分析：
计算出第一个分组的大小m，在正向遍历的过程中用cnt作为分割字符时的计数器，当cnt%k==0时添加分隔符-。
由于第一个分组大小不一定为k，将cnt初始化为-m。
具体步骤：
遍历字符串S获取有效字符长度n
根据分组情况计算出字节数组大小：size=n+n/k，size>0且n%k==0时，size减一
计数器cnt初始化为-n%k
正向遍历，cnt%k==0时添加分隔符
三、AC 代码：
func licenseKeyFormatting(s string, k int) string {
n := 0
for i := range s {
if s[i] != '-' {
n++
}
}
size := n + n/k
if size > 0 && n%k == 0 {
size--
}
ans := make([]byte, 0, size)
cnt := -n % k
for i := range s {
if s[i] == '-' {
continue
}
if s[i] >= 'a' && s[i] <= 'z' {
ans = append(ans, s[i]-'a'+'A')
} else {
ans = append(ans, s[i])
}
cnt++
if cnt%k == 0 && len(ans) < size {
ans = append(ans, '-')
}
}
return string(ans)
}
