LeetCode 541. 反转字符串 II Python

给定一个字符串和一个整数 k，你需要对从字符串开头算起的每个 2k 个字符的前k个字符进行反转。如果剩余少于 k 个字符，则将剩余的所有全部反转。如果有小于 2k 但大于或等于 k 个字符，则反转前 k 个字符，并将剩余的字符保持原样。

示例:

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

要求:

1. 该字符串只包含小写的英文字母。
2. 给定字符串的长度和 k 在[1, 10000]范围内。

class Solution:
    def reverseStr(self, s, k):
        n = len(s)
        res = ""
        temp = 0
        while n >= temp + 2 * k:
            res = res + s[temp:temp+k][::-1] + s[temp+k:temp+2*k]
            temp += 2 * k
        if n - temp < k:
            res += s[temp:][::-1]
        else:
            res += s[temp:temp+k][::-1] + s[temp+k:]
        return res