持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第13天,点击查看活动详情
一、题目描述:
给定一个许可密钥字符串 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
二、思路分析:
思路比较简单,主要是实现过程有点磕绊
- 计算出除了破折号之外的字符的个数lenOutOfDash
- 将lenOutOfDash个字符分成part份,其中第一份比较特殊,有first个,需要单独拿出来;其他每一份都有k个
- 遍历字符串
- 跳过'-'
- 第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;
}