有一个密钥字符串 S ,只包含字母,数字以及 '-'(破折号)。其中, N 个 '-' 将字符串分成了 N+1 组。
给你一个数字 K,请你重新格式化字符串,使每个分组恰好包含 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 个字符。
思路: 先是对于原字符串的处理, 1. 去除破折号’-‘;2. 将小写的字母转换为大写.
然后对于处理后的字符串进行增加破折号的操作;1. 若长度正好是k的整数倍;可以正常增加’-‘(每k个字符增加). 2. 若长度不是k的整数倍, 需要对于第一组字符另外处理。
string licenseKeyFormatting(string s, int k) {
string temp, result;
for(auto & c: s){
if(c!='-')
{
if(c>='a' && c<='z'){
temp += c - 'a' + 'A';
continue;
}
temp += c;
}
}
int n = temp.size();
if(n%k != 0){ // 这是处理第一组的特殊情况
for(int i = 0;i<n%k;i++){
result += temp[i];
}
if(n/k > 0){
result += '-';
}
}
for(int i = n%k,j=0; i<n; i++){// 除去了第一个
j++;
result += temp[i];
if(i!=n-1 && j%k == 0) result += '-';
}
return result;
}