LeetCode刷题记录——第541题（反转字符串Ⅱ）

题目描述

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

示例:

输入: s = “abcdefg”, k = 2
输出: “bacdfeg”

思路分析

• 如果字符串s长度小于k，那么直接反转整个字符串
• 如果字符串长度大于k且小于2k，则反转k之前的字符
• 如果字符串长度大于2k，那么就需要分步骤迭代进行

代码示例

class Solution(object):
    def reverseStr(self, s, k):
        """
        :type s: str
        :type k: int
        :rtype: str
        """
        if len(s) <= k:
            return s[::-1]
        elif len(s) > k and len(s) < 2*k:
            return s[0:k][::-1] + s[k:]
        else:
            return s[0:k][::-1] + s[k:2*k] + self.reverseStr(s[2*k:],k)