LeetCode541（力扣541）：反转字符串Ⅱ(C++)

541.反转字符串Ⅱ

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

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

示例:

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

提示：

该字符串只包含小写英文字母。
给定字符串的长度和 k 在 [1, 10000] 范围内。

我的题解

class Solution {
public:
    string reverseStr(string s, int k) {
        size_t n = s.size();
        size_t start = 0;
        size_t end = k - 1;
        size_t last = 2 * k - 1;
        
        if (n < k) {  //在字符串长度比k还小的时候，直接反转字符串
            size_t end = n - 1;
            while (start<end)
            {
                swap(s[start], s[end]);
                start++;
                end--;
            }
            return s;
        }

        while (last <= n+2*k ){  //在最后一段的情况是2*k的长度可能会超过字符船长度n，但是一定超过不了n+2*k
            if (end >= n - 1) {//这种情况是，例如要k=39，但是最后一段只剩下22个字符<39，就要对剩下的字符全部反转
                end = n - 1;
            }
            while (start<end) //这种情况是，还剩下不用反转的字符，只用反转需要反转的字符。
            { 
                swap(s[start], s[end]);
                start++;
                end--;
            }
            start = last + 1;
            end = last + k;
            last += 2 * k;
        }
        return s;
    }
};