给定一个字符串和一个整数 k,你需要对从字符串开头算起的每个 2k 个字符的前k个字符进行反转。如果剩余少于 k 个字符,则将剩余的所有全部反转。如果有小于 2k 但大于或等于 k 个字符,则反转前 k 个字符,并将剩余的字符保持原样。
示例:
输入: s = "abcdefg", k = 2 输出: "bacdfeg"
要求:
- 该字符串只包含小写的英文字母。
- 给定字符串的长度和 k 在[1, 10000]范围内。
【分析】
这道题的意思其实是每隔k隔字符,翻转k个字符,最后如果不够k个了的话,则全部翻转。比较直接的方法就是先用n/k算出来原字符串s能分成几个长度为k的字符串,然后开始遍历这些字符串,遇到2的倍数就翻转,翻转的时候注意考虑下是否已经到s末尾了,参见代码如下:
class Solution {
public:
string reverseStr(string s, int k) {
int n = s.size();
int count = n/k;
for(int i = 0; i <= count; ++i){
if( i % 2 == 0){
if(i * k + k < n)
reverse(s.begin() + i * k,s.begin() + i * k + k);
else
reverse(s.begin() + i * k,s.end());//到达末尾
}
}
return s;
}
};