541. 反转字符串 II【C++】

题目链接

题目
给定一个字符串 s 和一个整数 k，从字符串开头算起，每计数至 2k 个字符，就反转这 2k 字符中的前 k 个字符。

如果剩余字符少于 k 个，则将剩余字符全部反转。
如果剩余字符小于 2k 但大于或等于 k 个，则反转前 k 个字符，其余字符保持原样。

示例 1：
输入：s = “abcdefg”, k = 2
输出：“bacdfeg”
示例 2：
输入：s = “abcd”, k = 2
输出：“bacd”

提示：
1 <= s.length <= 10⁴
s 仅由小写英文组成
1 <= k <= 10⁴

思路

1. 为了方便，我们在遍历字符串时，以2k为步长进行遍历，如果步长为1，我们还需要单独判断当前位置是不是2k整数倍。
2. 对于小于k个字符，和大于等于k但小于2k的情况只需要使用以下判断

if (i + k < s.size())

3. 对于翻转字符串，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;
    }
};