算法题
字符串(string)
反转字符串II

反转字符串II

题目

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

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

示例 1:

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

示例 2:

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

思路

在遍历字符串的过程中,只要让 i += (2 * k),i 每次移动 2 * k 就可以了,然后判断是否需要有反转的区间。

解法

func reverseStr(s string, k int) string {
    b := []byte(s)
    // 从头开始,每2k个字符,反转前k个字符
    for i := 0; i < len(b); i += 2 * k {
        // 如果剩余字符大于等于k个,则将前k个字符反转
        if i + k <= len(b) {
            reverse(b[i:i+k])
        // 如果剩余字符少于k个,则将剩余字符全部反转
        } else {
            reverse(b[i:length])
        }
    }
    return string(b)
}
 
func reverse(b []byte) {
    left := 0
    right := len(b) - 1
    for left < right {
        b[left], b[right] = b[right], b[left]
        left++
        right--
    }
}