📃 题目描述

题目链接：541. 反转字符串 II - 力扣（LeetCode） (leetcode-cn.com) (opens new window)

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

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

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

示例 1：

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

示例 2：

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

🔔 解题思路

定义一个 reverse(str, left, right) 反转字符串 str 从 left 到 right 的字符串，然后循环调用这段反转的逻辑就行了。

循环的过程中，我们需要不断更新 left 和 right 的定义，其实也很简单，在遍历字符串的过程中，只要让 i += (2 * k)，i 每次移动 2 * k 就可以了，i 就是 left，然后 right 就是 (left + k - 1)

class Solution {
    public String reverseStr(String s, int k) {
        StringBuilder str = new StringBuilder(s);
        
        // 每计数至 2k 个字符，就反转这 2k 字符中的前 k 个字符
        for (int i = 0; i < str.length(); i += (2 * k)) {
            // 如果剩余字符小于 2k 但大于或等于 k 个，则反转前 k 个字符，其余字符保持原样
            if (i + k <= str.length()) {
                reverse(str, i, i + k - 1);
            }
            // 如果剩余字符少于 k 个，则将剩余字符全部反转
            else  {
                reverse(str, i, str.length() - 1);
            }
        }

        return str.toString();
    }

    // 反转 str[left, right]
    private void reverse(StringBuilder str, int left, int right) {
        while (left < right) {
            char tempLeft = str.charAt(left);
            char tempRight = str.charAt(right);
            str.setCharAt(left, tempRight);
            str.setCharAt(right, tempLeft);
            left ++;
            right --;
        }
    }
}

💥 复杂度分析

• 空间复杂度：O(N)
• 时间复杂度：O(N)