541. 反转字符串 II
Difficulty: 简单
给定一个字符串 s
和一个整数 k
,你需要对从字符串开头算起的每隔 2k
个字符的前 k
个字符进行反转。
- 如果剩余字符少于
k
个,则将剩余字符全部反转。 - 如果剩余字符小于
2k
但大于或等于k
个,则反转前k
个字符,其余字符保持原样。
示例:
输入: s = "abcdefg", k = 2
输出: "bacdfeg"
提示:
- 该字符串只包含小写英文字母。
- 给定字符串的长度和
k
在[1, 10000]
范围内。
Solution
注意两个边界条件的控制:
- 每隔
2k
个字符的前k
个字符进行反转,意味着第一个k间距反转,第二个k间距不反转,奇偶 - 剩余字符的处理问题,题目说得很懵,一不小心就被绕进去了,如果剩余字符串不足2k个,对
[i, min(i+k-1, len(s)-1)]
做反转
class Solution:
def reverseStr(self, s: str, k: int) -> str:
l = list(s)
for i, _ in enumerate(l):
if (i / k) % 2 == 0:
l = self.reverse(l, i, min(i+k-1, len(s)-1))
else:
continue
return "".join(l)
def reverse(self, arr, start, end):
while start < end:
arr[start], arr[end] = arr[end], arr[start]
start += 1
end -= 1
return arr