博客

1. 题目描述

题目链接：

541. 反转字符串 II

2. 解题思路

这道题是 反转字符串 的进阶版，看题目可能有点难理解，其实很简单。

我们以字符串

abcdefgz

为例，假设

k = 3

，如下：

其实在遍历字符串的过程中，只要让

i += (2 * k)

，

i

每次移动

2 * k

就可以了，然后判断是否需要有反转的区间。

因为要找的也就是每

2 * k

区间的起点，这样写，程序会高效很多。

所以当需要固定规律一段一段去处理字符串的时候，要想想在

for

循环的表达式上做做文章。

那么这里具体反转的逻辑我们要不要使用库函数呢，其实用不用都可以，使用

reverse

来实现反转也没毛病，毕竟不是解题关键部分。

3. 代码实现

使用 C++ 库函数

reverse

的版本如下：

class Solution {
public:
    string reverseStr(string s, int k) {
        for (int i = 0; i < s.size(); i += (2 * k)) {
            // 1. 每隔 2k 个字符的前 k 个字符进行反转
            // 2. 剩余字符小于 2k 但大于或等于 k 个，则反转前 k 个字符
            if (i + k <= s.size()) {
                reverse(s.begin() + i, s.begin() + i + k);
            }
            else {
                // 3. 剩余字符少于 k 个，则将剩余字符全部反转。
                reverse(s.begin() + i, s.end());
            }
        }
        return s;
    }
};

下面我实现的

reverse

函数区间是左闭右闭区间，代码如下：

class Solution {
public:
    //自己实现反转函数
    void myReverse(string& s, int start, int end) {
        for (int i = start, j = end; i < j; ++i, --j) {
            swap(s[i], s[j]);
        }
    }
    string reverseStr(string s, int k) {
        for (int i = 0; i < s.size(); i += (2 * k)) {
            if (i + k <= s.size()) {
                myReverse(s, i, i + k - 1);
            }
            else {
                myReverse(s, i, s.size() - 1);
            }
        }
        return s;
    }
};