题目

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

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

示例 1:

1
2
输入：s = "abcdefg", k = 2
输出："bacdfeg"

示例 2：

1
2
输入：s = "abcd", k = 2
输出："bacd"

提示：

1 <= s.length <= 104
s 仅由小写英文组成
1 <= k <= 104

解题思路：

其实这题很简单，我们只要改变一下思路就行

题目说"如果剩余字符小于 2K 但大于等于 K，则反转前 K，其余字符保持原样"

那么我们可以将这句话理解成，若剩余字符大于等于 2K 则我们就将 2K 的起点 + K 反转，其余的保持原样

代码：

Python

1
2
3
4
5
6
7
8
9
10
11
class Solution:
    def reverseStr(self, s: str, k: int) -> str:
        n = len(s)
        for idx in range(0, n, k * 2):
            # 将每个 2*k 的起点反转
            if idx + k <= n:
                s = s[: idx] + "".join(reversed(s[idx: idx + k])) + s[idx + k:]
                continue
            # 剩余字符少于 K 个，直接反转剩余的字符串即可
            s = s[: idx] + "".join(reversed(s[idx:]))
        return s

C++

1
2
3
4
5
6
7
8
9
10
11
12
13
14
class Solution {
public:
    string reverseStr(string s, int k) {
        int n = s.size();
        for (int i = 0; i < n; i += (2 * k)) {
            if (i + k <= s.size()) {
                reverse(s.begin() + i, s.begin() + i + k);
                continue;
            }
            reverse(s.begin() + i, s.end());
        }
        return s;
    }
};