LeetCode#482. 密钥格式化

114 阅读2分钟

一起养成写作习惯!这是我参与「掘金日新计划 · 4 月更文挑战」的第10天,点击查看活动详情

1.描述

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

2.分析

  1. 先将原字符串中的'-'去掉得到新字符串s

  2. 新字符串s长度对k取余数res:

    若res为0,则遍历字符串s,将每个元素添加进输出字符串,每遍历k个元素则加上'-';

    若res不为0,则先取字符串s前 res个元素添加进输出字符串,再加上'-',然后按着res=0,进行遍历

  3. 对得到的输出字符去掉最后一位(最后一位为多于的'-'),再用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)