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

95 阅读1分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第16天,点击查看活动详情

一、题目描述:

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

二、思路分析:

  • 将所有小写字母转化为大写字母,新的字符串命名为up_s;
  • 统计非"-"字符的个数,计为lens;
  • 将字符串中所有的"-"替换为"",新的字符串命名为ln_up_s;
  • 计算lens % k,如果有余数,说明第一个分组的个数小于k;反之,则第一个分组的个数等于k;
  • 再遍历ln_up_s,以k为长度,k为幅度,余数为起点,添加到结果res中。
  • 去掉res的最后一个“-”。

三、AC 代码:

class Solution:
    def licenseKeyFormatting(self, s: str, k: int) -> str:
        up_s = s.upper()
        # print(up_s)
        lens = len(s) - s.count("-")
        # print(lens)
        remainder = lens % k
        ln_up_s = up_s.replace("-", "")
        # print(ln_up_s)
        if remainder == 0:
            res = ""
        else:
            res = ln_up_s[:remainder] + "-"
        for i in range(remainder, lens, k):
            res += ln_up_s[i:i + k]
            res += "-"

        return res[:-1]

四、参考:

密钥格式化(c++与字符串解题) - 密钥格式化 - 力扣(LeetCode)