1. 问题描述:
有一个密钥字符串 S ,只包含字母,数字以及 '-'(破折号)。其中, N 个 '-' 将字符串分成了 N+1 组。给你一个数字 K,请你重新格式化字符串,使每个分组恰好包含 K 个字符。特别地,第一个分组包含的字符个数必须小于等于 K,但至少要包含 1 个字符。两个分组之间需要用 '-'(破折号)隔开,并且将所有的小写字母转换为大写字母。给定非空字符串 S 和数字 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 的长度可能很长,请按需分配大小。K 为正整数。
- S 只包含字母数字(a-z,A-Z,0-9)以及破折号'-'
- S 非空
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/license-key-formatting
2. 思路分析:
一个比较容易想到的方法是逆序遍历字符串,使用循环对当前位置往前找k个不是"-"的字符之后拼接到结果集中,break然后继续下一个k个字符的寻找,一开始的时候我们需要使用strip方法去除掉字符串开始与结束的"-"字符,这样可以避免"-"字符对结果造成影响,主要是模拟整个的过程即可
3. 代码如下:
class Solution:
def licenseKeyFormatting(self, S: str, K: int) -> str:
S = S.strip("-")
# 可以逆序遍历字符串按照K个字符进行拼接
res, i = "", len(S) - 1
while i >= 0:
count = 0
t = ""
while i >= 0 and count < K:
c = S[i]
# 将小写字母换成大写字母
if "a" <= c <= "z": c = c.upper()
if c != "-":
t = c + t
count += 1
# 注意最后一个字符的下标
if count == K: break
i -= 1
res = "-" + t + res + ""
i -= 1
# 截取第二个开始的字符串
return res[1:]