字符串 | LeetCode 541 反转字符串 II

36 阅读1分钟

题目描述

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

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

本题思路

class Solution {
public:
    string reverseStr(string s, int k) {
        for (int i = 0; i < s.size(); i += 2 * k) {
            if (i + k <= s.size()) {
                reverse(i, i + k - 1, s);
                continue;
            }
            reverse(i, s.size() - 1, s);
        }
        return s;
    }
    void reverse(int m, int n, string& s) {
        for (int i = m, j = n; i < j; i++, j--)
            swap(s[i], s[j]);
    }
};

分三种情况,被k和2k分为三个区间。

第一个区间内需要翻转0 ~ end。

k ~ 2k和2k之后,两个区间都是要翻转0 ~ k,可以合并为一种。

最后传参时,一定要带&,否则原本的值根本没有更改。