541.反转字符串II
给定一个字符串 s 和一个整数 k,你需要对从字符串开头算起的每隔 2k 个字符的前 k 个字符进行反转。
如果剩余字符少于 k 个,则将剩余字符全部反转。
如果剩余字符小于 2k 但大于或等于 k 个,则反转前 k 个字符,其余字符保持原样。
class Solution {
public:
string reverseStr(string s, int k) {
if (s.size() <= 1) return s;
int i;
for (i = 0; i + 2 * k < s.size(); i += 2 * k) {
reverse(s.begin()+i,s.begin()+i+k);
}
// i -= 2 * k;
if (i != s.size() - 1) {
if (i + k < s.size()) {
reverse(s.begin()+i,s.begin()+i+k);
}
else reverse(s.begin()+i,s.end());
}
return s;
}
};