一起养成写作习惯!这是我参与「掘金日新计划 · 4 月更文挑战」的第10天,点击查看活动详情。
1.描述
给定一个许可密钥字符串 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
2.分析
-
先将原字符串中的'-'去掉得到新字符串s
-
新字符串s长度对k取余数res:
若res为0,则遍历字符串s,将每个元素添加进输出字符串,每遍历k个元素则加上'-';
若res不为0,则先取字符串s前 res个元素添加进输出字符串,再加上'-',然后按着res=0,进行遍历
-
对得到的输出字符去掉最后一位(最后一位为多于的'-'),再用upper函数调整大小写,最后输出字符串
3.AC代码
class Solution(object):
def licenseKeyFormatting(self, s, k):
"""
:type s: str
:type k: int
:rtype: str
"""
s = s.replace('-','')
s_len = len(s)
out_key = ''
res = s_len % k
if res == 0:
for id , char in enumerate(s):
out_key += char
if (id+1)% k == 0:
out_key += '-'
else:
out_key = s[:res]+'-'
for id,char in enumerate(s[res:]):
out_key += char
if (id+1)% k == 0:
out_key += '-'
return out_key[:-1].upper()
参考
清晰易懂的两种解法(击败97%), 含详细注释【大白技术控】 - 密钥格式化 - 力扣(LeetCode) (leetcode-cn.com)