大家好，我是木川

一、题目描述

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

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

示例 1：

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

示例 2：

输入：s = "abcd", k = 2
输出："bacd"

提示：

  • 1 <= s.length <= 104
  • s 仅由小写英文组成
  • 1 <= k <= 104

二、题解

一）遍历+双指针

思路：每遍历 2k 个字符，使用双指针反转长度为 k 的子串

代码：

func reverseStr(s string, k int) string {
    b := []byte(s)
    // 从每个下标从2k的倍数开始
    for i := 0; i < len(s); i += 2 * k {
        // 反转长度为k的子串
        left := i 
        right := i + k - 1
        // 越界判断
        if right > len(s)-1 {
            right = len(s)-1
        }
        for left < right  {
            b[left], b[right] = b[right], b[left]
            left++
            right--
        }
    }  

    return string(b)
}

时间复杂度：O(n)，外层循环遍历时间复杂度 O(n/2k),  内层循环遍历时间复杂度O(n/2)，综合时间复杂度 O(n/2k) * O(n/2) = O(n)

空间复杂度：O(1)，只使用了常数空间来存放若干变量