【leetcode题解C++】541.反转字符串II

541. 反转字符串II

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

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

示例 1：

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

示例 2：

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

思路1：根据344.反转字符串，以及本题用到了STL，第一时间想到了reverse()库函数，但是在344题的题解中建议我们不要使用库函数，那么swap()。用一个tmp字符串累计存储2k个字符，处理完过后，加入ans字符串，接下来的话，就是一个很暴力的解法了。

需要注意的就是swap()中下标不要写错了。

代码实现1：

class Solution {
public:
    string reverseStr(string s, int k) {
        string ans;
        int p = 0;
        int round = 0;
        while (p < s.length()) {
            string tmp;
            while ((p - 2 * k * round) < 2 * k) {
                if (p < s.length()) {
                    tmp.push_back(s[p]);
                    ++p;
                } else {
                    break;
                }
            }
            ++round;
            if (tmp.length() < k) {
                for (int i = 0; i < tmp.length() - i; ++i) {
                    swap(tmp[i], tmp[tmp.length() - i - 1]);
                }
            }
            else if (k <= tmp.length() < 2*k) {
                for (int i = 0; i < k - i - 1; ++i) {
                    swap(tmp[i], tmp[k - i - 1]);
                }
            }
            else if (tmp.length() == 2*k) {
                for (int i = 0; i < k - i - 1; ++i) {
                    swap(tmp[i], tmp[k - i -1]);
                }
            }
            ans.append(tmp);
        }
        return ans;
    }
};

思路2：官方题解中使用了reverse()...代码量少了很多，这让作者很高兴啊hah...

代码实现2：

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;
    }
};

?