题目描述:
有一个密钥字符串 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 个字符。
提示:
S 的长度可能很长,请按需分配大小。K 为正整数。
S 只包含字母数字(a-z,A-Z,0-9)以及破折号’-’
S 非空
代码:
class Solution {
public:
string licenseKeyFormatting(string s, int k) {
int l = s.length(),num=0;
string ss ="";
for(int i=l-1;i>=0;i--){
if(s[i]!='-'){
if(islower(s[i])) //是小写字母
s[i]=toupper(s[i]); //转为大写字母
ss.push_back(s[i]);
num++;
}
if(num==k&&i!=0){
ss.push_back('-');
num=0;
}
}
for(int i=ss.length()-1;i>=0;i--){ //防止输入的字符串以很多-开头,防止最终结果的开头有-,先去掉多余的-,再反转
if(ss[i]=='-')ss=ss.replace(i,1,"");
else break;
}
reverse(ss.begin(),ss.end());
return ss;
}
};
题目要求第一组可以小于等于k,即第一组字符的个数相当于l%k(字符串中非-字符的总长度对k取余数)。所以从字符串末尾从后向前遍历。
因为从后向前遍历,最后要反转把顺序变正。每次取k个字符,最后剩下的一组自然个数小于等于k,最后反转得到结果。
注意:
拼接字符串要用str.push_back(char),不能直接用str=str+s,避免超时。