题目描述
给定一个字符串 s 和一个整数 k,从字符串开头算起,每计数至 2k 个字符,就反转这 2k 字符中的前 k 个字符。
如果剩余字符少于 k 个,则将剩余字符全部反转。
如果剩余字符小于 2k 但大于或等于 k 个,则反转前 k 个字符,其余字符保持原样。
思路与算法
- 遍历字符串:
从头到尾遍历字符串,以 2k 为步长进行划分 - 边界条件:
如果剩余字符小于 k,则需要反转所有剩余字符。
如果剩余字符大于等于 k 但小于 2k,则反转前 k 个字符,剩下的不变。 - 实现
s[start:end] 用来选取子字符串
使用 [::-1] 来反转
代码
class Solution:
def reverseStr(self, s: str, k: int) -> str:
res = list(s) # 将字符串转换为列表,便于修改
n = len(s)
# 以 2k 为步长
for i in range(0, n, 2*k):
res[i:i+k] = res[i:i+k][::-1]
return ''.join(res)
总结
- 由于字符串是不可变的,为了方便修改,我们将字符串转换为列表 result = list(s),修改完成后再转换回字符串。
- 使用切片操作 s[start:end] 选取子字符串,结合反转操作 [::-1](切片操作的一种特殊情况),可以高效地反转字符串的指定部分。