Skip to content

541.反转字符串II

https://leetcode.cn/problems/reverse-string-ii/

给定一个字符串 s 和一个整数 k,从字符串开头算起,每计数至 2k 个字符,就反转这 2k 字符中的前 k 个字符。

  • 如果剩余字符少于 k 个,则将剩余字符全部反转。
  • 如果剩余字符小于 2k 但大于或等于 k 个,则反转前 k 个字符,其余字符保持原样。

示例 1:

输入:s = "abcdefg", k = 2
输出:"bacdfeg"

示例 2:

输入:s = "abcd", k = 2
输出:"bacd"

提示:

  • 1 <= s.length <= 104
  • s 仅由小写英文组成
  • 1 <= k <= 104
python
class Solution:
    def reverseStr(self, s: str, k: int) -> str:
        s = list(s)  # 将字符串转为列表,方便修改字符
        n = len(s)
        
        for i in range(0, n, 2 * k):
            # 判断剩余字符的长度
            if i + k <= n:
                s[i:i+k] = reversed(s[i:i+k])  # 反转前 k 个字符
            else:
                s[i:n] = reversed(s[i:n])  # 剩余字符少于 k 个,直接反转
            
        return ''.join(s)  # 将列表转换回字符串