541. 反转字符串 IIhttps://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
实现:
class Solution:def reverseStr(self, s: str, k: int) -> str:length = len(s)s_list = list(s)times = length // (2 * k)remain = length - 2 * k * timesindex = 0for i in range(times):left = indexright = left + k - 1while left < right:tmp = s_list[left]s_list[left] = s_list[right]s_list[right] = tmpleft += 1right -= 1index += 2 * kleft = indexif remain < k:right = length - 1else:right = left + k - 1while left < right:tmp = s_list[left]s_list[left] = s_list[right]s_list[right] = tmpleft += 1right -= 1s = "".join(s_list)return s
优化实现:
-
每
2k
个字符为一组,反转其中的前k
个字符。 -
如果剩余字符不足
k
个,则反转所有剩余字符
class Solution:def reverseStr(self, s: str, k: int) -> str:s_list = list(s)for i in range(0, len(s), 2*k):left = iif i + k <= len(s_list):right = i + k - 1self.reverse(s_list, left, right)else:right = len(s_list) - 1self.reverse(s_list, left, right)return ''.join(s_list)def reverse(self, s_list: List[str], left: int, right: int ) -> None:while left < right:tmp = s_list[left]s_list[left] = s_list[right]s_list[right] = tmpleft += 1right -= 1