Leetcode 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 <= 104
• s 仅由小写英文组成
• 1 <= k <= 104

思路:

每次移动2k个单位，前k个字符反转。遍历length / 2*k次，每次指针指向前k个字符的头和尾。

处理尾部，长度小于k直接从n*2*k开始，length-1结束设置指针。

在k与2*k之间，从n*2*k开始，n*2*k+k-1结束。

解题过程:

求遍历次数。声明指针变量。

int length = s.size();
int times = length / (2 * k);
int start, end;

 开始遍历，每次移动2k个单位。反转前k个字符。

for(int i = 0; i < times; i++){
    start = i * 2 * k;
    end = i * 2 * k + k - 1;
    while(start < end){
        swap(s[start], s[end]);
        start++;
        end--;
    }
}

处理结尾。

if(length % (2 * k) >= k){
    start = times * 2 * k;
    end = times * 2 * k + k - 1;
    while(start < end){
        swap(s[start], s[end]);
        start++;
        end--;
    }
}else{
    start = times * 2 * k;
    end = length - 1;
    while(start < end){
        swap(s[start], s[end]);
        start++;
        end--;
    }
}
return s;

题解:

class Solution {
public:

    string reverseStr(string s, int k) {
        int length = s.size();
        int times = length / (2 * k);
        int start, end;
        for(int i = 0; i < times; i++){
            start = i * 2 * k;
            end = i * 2 * k + k - 1;
            while(start < end){
                swap(s[start], s[end]);
                start++;
                end--;
            }
        }
        if(length % (2 * k) >= k){
            start = times * 2 * k;
            end = times * 2 * k + k - 1;
            while(start < end){
                swap(s[start], s[end]);
                start++;
                end--;
            }
        }else{
            start = times * 2 * k;
            end = length - 1;
            while(start < end){
                swap(s[start], s[end]);
                start++;
                end--;
            }
        }
        return s;
    }
};