541. 反转字符串 II
给定一个字符串 s
和一个整数 k
,你需要对从字符串开头算起的每隔 2k
个字符的前 k
个字符进行反转。
- 如果剩余字符少于
k
个,则将剩余字符全部反转。 - 如果剩余字符小于
2k
但大于或等于k
个,则反转前k
个字符,其余字符保持原样。
示例:
输入: s = "abcdefg", k = 2
输出: "bacdfeg"
暴力
class Solution {
public:
string reverseStr(string s, int k) {
if(s == "") return s;
if(s.size() <= k) {
reverse(s.begin(), s.end());
return s;
}
for (int start = 0; start < s.size(); start += k * 2) {
int i = start;
int j = min<int>(start + k - 1, s.size() - 1);
while (i < j) {
char tmp = s[i];
s[i++] = s[j];
s[j--] = tmp;
}
}
return s;
}
};