541. 反转字符串 II(字符串)(Leetcode刷题笔记)
https://lunan0320.cn
文章目录
题目
给定一个字符串 s 和一个整数 k,从字符串开头算起,每计数至 2k 个字符,就反转这 2k 字符中的前 k 个字符。
如果剩余字符少于 k 个,则将剩余字符全部反转。
如果剩余字符小于 2k 但大于或等于 k 个,则反转前 k 个字符,其余字符保持原样。
解题代码 C++
class Solution {
public:
//局部交换的函数 从begin 到 end (闭区间)逆序
void reverse_substr(string& s, int begin, int end){
while(end > begin){
//使用swap函数
swap(s[begin], s[end]);
begin++;
end--;
}
}
string reverseStr(string s, int k) {
int length = s.size();
if (length <= 1) return s;
int i = 0;
while (i < length){
//i 到i+k闭区间的翻转
// 判断剩余部分不足k个的直接全部翻转
if (i + k - 1 >= length) reverse_substr(s, i, s.size()-1);
else reverse_substr(s, i, i+k-1);
//i每次需要递增2k
i = i + 2*k;
}
return s;
}
};