leetcode算法482. 密钥格式化

82 阅读1分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第13天,点击查看活动详情

一、题目描述:

482. 密钥格式化 - 力扣(LeetCode)

给定一个许可密钥字符串 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

二、思路分析:

思路比较简单,主要是实现过程有点磕绊

  1. 计算出除了破折号之外的字符的个数lenOutOfDash
  2. 将lenOutOfDash个字符分成part份,其中第一份比较特殊,有first个,需要单独拿出来;其他每一份都有k个
  3. 遍历字符串
    • 跳过'-'
    • 第0个part只有first个,如果字符串只能分为一部分,即part=1,则无需'-';如果part>1,增加'-'
    • 第1~part-1个有k个字符串,需要注意的是第part-1个字符串后不需要'-'

三、AC 代码:

char convert(char a){
    char ans = a;
    if(a >= 'a' && a <= 'z'){
        ans = a - 'a' + 'A';
    }
    return ans;
}

char * licenseKeyFormatting(char * s, int k){
    char * ans = malloc(sizeof(char) * strlen(s) * 2);
    int lenOutOfDash = 0;
    for(int i = 0; i < strlen(s); i++){
        if(s[i] != '-'){
            lenOutOfDash++;
        }
    }
    int part = lenOutOfDash / k + (lenOutOfDash % k != 0);
    int first = lenOutOfDash % k != 0 ? lenOutOfDash - lenOutOfDash / k * k : k;
    int len = 0, j = 0, m = 0;
    for(int i = 0; i < strlen(s); i++){
        if(s[i] == '-'){
            continue;
        }
        ans[len++] = convert(s[i]);
        j++;
        if((j == first && m == 0 && part > 1) || (j == k && m > 0 && m < part - 1)){
            ans[len++] = '-';
            j = 0;
            m++;
        }
    }
    ans[len] = '\0';

    return ans;
}