541. 反转字符串II

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

如果剩余字符少于 k 个，则将剩余字符全部反转。

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

示例:

输入: s = "abcdefg", k = 2
输出: "bacdfeg"

 

class Solution {
public:
    string reverseStr(string s, int k) {
        int len = s.length();
        if (!len) return s;
        int n = len / k;
        int num = len / (2 * k);
        int last = len - len % (2 * k);
        vector<int> vec;
        int index = 0;
        vec.push_back(last);
        //保存要处理的下标
        for (int i = 0; i < num; i++)
        {
            vec.push_back(index);
            index = index + 2 * k;
            if (index == last)
            {
                break;
            }
        }
        //反转下标对应的string
        for (const auto p : vec)
        {
            int left = p;
            int right = left + k - 1;
            if (right > len - 1) right = len - 1;
            char cur;
            while (left < right)
            {
                cur = s[left];
                s[left] = s[right];
                s[right] = cur;
                left++;
                right--;
            }
        }
        return s;
    }
};