【leetcode】541. 反转字符串 II

文章目录

      • 代码
        • 1. 切片
        • 2. 转为list

给定一个字符串 s 和一个整数 k,从字符串开头算起,每计数至 2k 个字符,就反转这 2k 字符中的前 k 个字符。

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

示例 1:

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

输入:s = “abcd”, k = 2
输出:“bacd”

代码

1. 切片
class Solution(object):
    def reverseStr(self, s, k):
        """
        :type s: str
        :type k: int
        :rtype: str
        """
        p = 0
        while p < len(s):
            p2 = p + k
            s = s[:p] + s[p:p2][::-1] + s[p2:]
            p = p + 2 * k
        return s
        
2. 转为list
class Solution(object):
    def reverseStr(self, s, k):
        """
        :type s: str
        :type k: int
        :rtype: str
        """
        s_list = list(s)
        i = 0
        while i < len(s):
            s_list[i:i+k] = s_list[i:i+k][::-1]
            i = i + 2 * k
        return ''.join(s_list) 
        

参考:
代码随想录:https://programmercarl.com/

你可能感兴趣的:(LeetCode,leetcode,算法,职场和发展)