LeetCode刷题记录——第482题（密钥格式化）

题目描述

给定一个密钥字符串S，只包含字母，数字以及 ‘-’（破折号）。N 个 ‘-’ 将字符串分成了 N+1 组。给定一个数字 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个字符，所以insert的位置应该是len(s) - i

代码示例

class Solution(object):
    def licenseKeyFormatting(self, S, K):
        """
        :type S: str
        :type K: int
        :rtype: str
        """
        S = S.upper().replace('-','')
        res = list(S)
        for i in range(K,len(S),K):
            res.insert(len(S)-i,'-')
        return ''.join(res)