题目链接
题目
给定一个字符串 s 和一个整数 k,从字符串开头算起,每计数至 2k 个字符,就反转这 2k 字符中的前 k 个字符。
如果剩余字符少于 k 个,则将剩余字符全部反转。
如果剩余字符小于 2k 但大于或等于 k 个,则反转前 k 个字符,其余字符保持原样。
示例 1:
输入:s = “abcdefg”, k = 2
输出:“bacdfeg”
示例 2:
输入:s = “abcd”, k = 2
输出:“bacd”
提示:
1 <= s.length <= 104
s 仅由小写英文组成
1 <= k <= 104
思路
- 为了方便,我们在遍历字符串时,以2k为步长进行遍历,如果步长为1,我们还需要单独判断当前位置是不是2k整数倍。
- 对于小于k个字符,和大于等于k但小于2k的情况只需要使用以下判断
if (i + k < s.size())
- 对于翻转字符串,C++可以使用reverse函数(左闭右开区间),参考链接如下:
reverse函数简单使用
也可以自己写一个翻转字符串函数(左闭右闭区间)
void my_reverse(string s){
int n = s.size();
int left = 0;
int right = n - 1;
while (left <= right){
swap(s[left++], s[right--]);
}
}
整体代码
class Solution {
public:
string reverseStr(string s, int k) {
for (int i = 0; i < s.size(); i += 2 * k){ //关键2k
if (i + k < s.size()){
reverse(s.begin() + i,s.begin() + i + k);
}else{
reverse(s.begin() + i, s.end());
}
}
return s;
}
};