算法笔记 — 482. 密钥格式化

Offer 驾到，掘友接招！我正在参与2022春招打卡活动，点击查看活动详情。

一、题目描述：

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

二、思路分析：

• 使用ReplaceAll函数将“-”替换成“”，即去掉“-”
• 使用ToUpper函数将小写字母全部转换成大写
• 如果字符串的长度 n 刚好是 k 的倍数，即 n%k==0 ，那么依次从字符串中取出k个字符进行append，再加上“-”
• 如果n%k!=0,那么第一次append的时候只需要从字符串中取出前 n%k 个字符，再加上“-”，后面再依次取出k个
• 最后去除末尾的的“-”

三、AC 代码：

func licenseKeyFormatting(s string, k int) string {
	ss := strings.ToUpper(strings.ReplaceAll(s, "-", ""))
	n := len(ss)
	var ans []byte
	if n % k != 0 {
		ans = append(append(ans, ss[:n%k]...),'-')
	}
	for i := n%k; i < n; i+=k {
		ans = append(append(ans, ss[i:i+k]...),'-')
	}
	if len(ans) > 0 && ans[len(ans)-1] == '-' {
		ans = ans[:len(ans)-1]
	}
	return string(ans)
}
复制代码