反转字符串II

本文参考代码随想录

给定一个字符串 s 和一个整数 k，从字符串开头算起, 每计数至 2k 个字符，就反转这 2k 个字符中的前 k 个字符。
如果剩余字符少于 k 个，则将剩余字符全部反转。
如果剩余字符小于 2k 但大于或等于 k 个，则反转前 k 个字符，其余字符保持原样。

在遍历字符串时，每次移动2k

class Solution {
public:
    string reverseStr(string s, int k) {
        for(int i = 0; i < s.size();i += (2*k)){
            if(i + k <= s.size()){
                reverse(s.begin() + i, s.begin() + i + k);
            }else{
                reverse(s.begin() + i, s.end());
            }
        }
        return s;
    }
};

时间复杂度O(n)
空间复杂度O(1)

另一种思路：

class Solution {
public:
    string reverseStr(string s, int k) {
        int n = s.size(), pos = 0;
        while(pos < n){
            if(pos + k < n) reverse(s.begin() + pos, s.begin() + pos + k);
            else reverse(s.begin() + pos, s.end());
            pos += 2 * k;
        }
        return s;
    }
};

时间复杂度: O(n)
空间复杂度: O(1)