LeetCode 541. Reverse String II

题目描述

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

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

要求

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

算法

(模拟) O(n)
循环枚举每 2k 个字母的开头位置。

时间复杂度

每个字母最多仅被遍历一次，故时间复杂度为 O(n)。

python code

class Solution:
    def reverseStr(self, s: str, k: int) -> str:
        l = 0
        mid = k
        r = 2 * k
        res = ''
        while len(res) < len(s):
            res += s[l:mid][::-1] + s[mid:r]
            l = 2 * k + l
            mid = mid + 2 * k
            r = r + 2 * k
        return res