LeetCode541. 反转字符串 II（每隔 2k 个字符的前 k 个字符反转）

1、题目描述

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

给定一个字符串 s 和一个整数 k，你需要对从字符串开头算起的每隔 2k 个字符的前 k 个字符进行反转。

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

通俗一点说，每隔k个反转k个，末尾不够k个时全部反转

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

"ab cd ef g"
"ba cd fe g"

2、代码详解

双指针等距向后遍历

class Solution(object):
    def reverseStr(self, s, k):
        """
        :type s: str
        :type k: int
        :rtype: str
        输入: "ab cd ef g", k = 2
        输出: "ba cd fe g"
        """
        # 每隔k个反转k个，末尾不够k个时全部反转
        left, mid, right = 0, k, 2*k
        res = ''
        while len(res) < len(s):
            res += s[left:mid][::-1] + s[mid:right]
            left, mid, right = left + 2*k, mid + 2*k, right + 2*k
        return res

ss = "abcdefg"
k = 2
s = Solution()
print(s.reverseStr(ss, k))