1.题目描述
给定一个字符串 s 和一个整数 k,从字符串开头算起,每计数至 2k 个字符,就反转这 2k 字符中的前 k 个字符。
如果剩余字符少于 k 个,则将剩余字符全部反转。如果剩余字符小于 2k 但大于或等于 k 个,则反转前 k 个字符,其余字符保持原样。
示例 1:
输入:s = "abcdefg", k = 2
输出:"bacdfeg"
示例 2:
输入:s = "abcd", k = 2
输出:"bacd"
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/reverse-string-ii
2.解题思路
这一题我们在遍历字符串的时候,可以一次性让i变成i+2k,这时候判断剩下的字符有没有k个,如果有k个的话,我们就把前面2k个字符进行reverse,然后继续进行循环的continue,没有的话就把剩下的不足k个字符全部reverse。
3.代码实现
class Solution {
public:
string reverseStr(string s, int k) {
for(int i = 0; i < s.size(); i += 2*k)
{
if(i + k <= s.size())
{
reverse(s.begin() + i, s.begin() + i + k);
continue;
}
reverse(s.begin() + i, s.begin() + s.size());
}
return s;
}
};