541. 反转字符串 II【双指针】

541. 反转字符串 II

541. 反转字符串 II

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

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

示例 1：

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

示例 2：

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

提示：

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

Go 代码

func reverseStr(s string, k int) string {
    ss := []byte(s)
    length := len(s)

    // 每隔 2k 个字符作为一组
    for i := 0; i < length; i += 2 * k {
        // 当前组字符只要大于等于 k 个，则反转该组前 k 个字符
        if i + k <= length {
            reverse(ss[i:i+k])
        } else { // 剩余字符小于k个了，反转剩余的所有字符
            reverse(ss[i:length])
        }
    }
    return string(ss)
}

func reverse(b []byte) {
    left := 0
    right := len(b) - 1
    for left < right {
        b[left], b[right] = b[right], b[left]
        left++
        right--
    }
}