反转字符串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--
}
}