02 反转字符串II（leecode 541）

1 问题

给定一个字符串 s 和一个整数 k，你需要对从字符串开头算起的每隔 2k 个字符的前 k 个字符进行反转。

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

示例:

输入: s = “abcdefg”, k = 2
输出: “bacdfeg”

2 解法

class Solution {
public:
    //反转函数
    void reverse(string& s, int begin, int end)
    {
        int offset = (end - begin + 1) / 2;
        //begin+offset表示下一次的起始位置计算的中间值
        for(int i = begin, j = end; i < begin + offset; i++, j--)
        {
            int tmp = s[i];
            s[i] = s[j];
            s[j] = tmp;
        }
    }
    string reverseStr(string s, int k) {
        //i指向剩余元素的开头
        for(int i = 0; i < s.size(); i += 2*k)
        {
            //1.剩余字符大于2k，从开头算起将前k个字符反转
            //2.剩余字符小于2k，但大于等于k，将将前k个字符反转
            if(s.size() - (i+1) >= k)
            {
                reverse(s, i, i + k - 1);
                continue;
            }
            //3.剩余字符少于k，全部反转
            else
            {
                reverse(s, i, s.size() - 1);
            }
        }
        return s;
    }
};