541. 反转字符串 II（力扣LeetCode）

541. 反转字符串 II

题目描述

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

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

示例 1：

输入：s = “abcdefg”, k = 2
输出：“bacdfeg”

示例 2：

输入：s = “abcd”, k = 2
输出：“bacd”

提示：

• 1 <= s.length <= 10⁴
• s 仅由小写英文组成
• 1 <= k <= 10⁴

暴力

这段代码的工作流程分为两个阶段：

1. 对于字符串 s 的每个字符，使用一个循环和一个计数器 count 来跟踪当前的位置。每次 count 达到 2k 时，就反转 l 和 l+k-1 之间的字符，并将 l 调整到下一个要反转的部分的起始位置。

2. 循环完成后，根据剩余字符的数量决定最后的反转策略。如果剩余字符少于 k 个，则反转这些字符。如果剩余字符的数量在 k 到 2k 之间，则只反转前 k 个字符。

注意：reverse 是一个辅助私有函数，它在原字符串 s 上直接进行操作，实现局部反转。这符合题目要求的原地修改字符串，并且不使用额外的空间。

class Solution {
public:
    // 该函数接收一个字符串s和一个整数k，返回按题目要求反转的字符串。
    string reverseStr(string s, int k) {
        int l = 0; // l是每个子区间反转部分的起始索引。
        int count = 0; // count用于跟踪当前遍历的字符数量，以确定何时进行反转。

        // 遍历字符串s中的每个字符。
        for(int i = 0; i < s.size(); i++) {
            count++; // 对遍历的字符进行计数。
            // 当count达到2k时，说明到达了一个反转点。
            if(count == 2 * k) {
                // 反转从索引l到l+k-1的字符。
                reverse(s, l, l + k - 1);
                // 将l移动到下一个待处理区间的开始。
                l = l + 2 * k;
                // 重置count为0，开始新一轮的计数。
                count = 0;
            }
        }
        // 字符串遍历完成后，可能还有字符没有被处理。
        // 如果剩余字符少于k个，则将剩余字符全部反转。
        if(count < k) {
            reverse(s, l, s.size() - 1);
        }
        // 如果剩余字符小于2k但大于或等于k个，则反转前k个字符。
        else if(count >= k && count < 2 * k) {
            reverse(s, l, l + k - 1);
        }
        // 返回处理后的字符串。
        return s;
    }

private:
    // 辅助函数：反转字符串s中从索引l到r的部分。
    void reverse(string& s, int l, int r) {
        // 使用两个指针l和r进行遍历，从两端开始，向中间靠拢。
        while(l < r) {
            // 交换l和r指向的字符。
            swap(s[l], s[r]);
            // 将l向右移动，r向左移动。
            l++, r--;
        }
        // 当l和r相遇或者交错时，停止反转。
    }
};

优化：使用reverse函数

这道题目其实也是模拟，实现题目中规定的反转规则就可以了。

一些同学可能为了处理逻辑：每隔2k个字符的前k的字符，写了一堆逻辑代码或者再搞一个计数器，来统计2k，再统计前k个字符。

其实在遍历字符串的过程中，只要让 i += (2 * k)，i 每次移动 2 * k 就可以了，然后判断是否需要有反转的区间。

因为要找的也就是每2 * k 区间的起点，这样写，程序会高效很多。

class Solution {
public:
    // 该函数接收一个字符串s和一个整数k，根据题目规则返回反转后的字符串。
    string reverseStr(string s, int k) {
        // 从字符串开头开始，以步长为2k遍历字符串。
        for(int i = 0; i < s.size(); i += 2 * k) {
            // 检查当前位置加上k是否超出了字符串的长度。
            if(i + k <= s.size() - 1)
                // 如果没有超出长度，那么反转从当前位置i开始的k个字符。
                // std::reverse是C++ STL中的函数，它接受两个迭代器参数：
                // s.begin() + i 是子字符串起始位置的迭代器。
                // s.begin() + i + k 是子字符串结束位置的迭代器，反转不包括这个位置的字符。
                reverse(s.begin() + i, s.begin() + i + k);
            else
                // 如果当前位置加上k超出了字符串的长度，
                // 那么反转从当前位置i到字符串末尾的所有字符。
                // s.end() 是字符串末尾的迭代器。
                reverse(s.begin() + i, s.end());
        }
        // 返回修改后的字符串。
        return s;
    }
};