题目描述
给定一个字符串 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,可以合并为一种。
最后传参时,一定要带&,否则原本的值根本没有更改。