题目描述
给定一个字符串 s
和一个整数 k
,从字符串开头算起,每计数至 2k
个字符,就反转这 2k
字符中的前 k
个字符。
- 如果剩余字符少于
k
个,则将剩余字符全部反转。 - 如果剩余字符小于
2k
但大于或等于k
个,则反转前k
个字符,其余字符保持原样。
示例 1:
输入:s = "abcdefg", k = 2
输出:"bacdfeg"
示例 2:
输入:s = "abcd", k = 2
输出:"bacd"
提示:
1 <= s.length <= 10^4
s
仅由小写英文组成1 <= k <= 10^4
思路
本题可以用模拟来解决,反转可以直接用库函数。注意要让i每次移动2*k个位置,从而减少遍历的次数。
代码
C++版:
class Solution {
public:
string reverseStr(string s, int k) {
// 每次移动2*k
for(int i=0;i<s.size();i+=2*k){
if(i+k<=s.size()){
// 剩余字符小于 2k 但大于或等于 k 个,则反转前 k 个字符
reverse(s.begin()+i,s.begin()+i+k);
}else{
// 剩余字符少于 k 个,则将剩余字符全部反转
reverse(s.begin()+i,s.end());
}
}
return s;
}
};
Python版:
class Solution:
def reverseStr(self, s: str, k: int) -> str:
def reverse_substring(text):
left, right = 0, len(text) - 1
while left < right:
text[left], text[right] = text[right], text[left]
left += 1
right -= 1
return text
res = list(s)
for cur in range(0, len(s), 2 * k):
res[cur: cur + k] = reverse_substring(res[cur: cur + k])
return ''.join(res)
需要注意的地方
1.当需要固定规律一段一段去处理字符串的时候,要想办法在for循环上做文章。