这是数组的第7题,力扣链接。
给定一个整数数组
nums
,将数组中的元素向右轮转k
个位置,其中k
是非负数。示例 1:
输入: nums = [1,2,3,4,5,6,7], k = 3 输出: [5,6,7,1,2,3,4] 解释: 向右轮转 1 步: [7,1,2,3,4,5,6] 向右轮转 2 步: [6,7,1,2,3,4,5] 向右轮转 3 步: [5,6,7,1,2,3,4]示例 2:
输入:nums = [-1,-100,3,99], k = 2 输出:[3,99,-1,-100] 解释: 向右轮转 1 步: [99,-1,-100,3] 向右轮转 2 步: [3,99,-1,-100]
在做链表的旋转时其实做过类似的,不过当时的最优解是把链表做成环,这里当然也可以,但就不这样做了,因为逻辑可能会比较复杂,需要维护额外的指针。
看到这道题我的第一反应是拿切片做,因为本质是两个切片互换顺序,由于需要更改原数组因此这里需要一个额外的赋值操作。
func rotate(nums []int, k int) {
k %= len(nums)
result := make([]int, 0)
result = append(result, nums[len(nums)-k:]...)
result = append(result, nums[:len(nums)-k]...)
copy(nums, result)
}
还有一种思路是先将数组全反,然后旋转前k个和后k个。
func reverse(a []int) {
for i, n := 0, len(a); i < n/2; i++ {
a[i], a[n-1-i] = a[n-1-i], a[i]
}
}
func rotate(nums []int, k int) {
k %= len(nums)
reverse(nums)
reverse(nums[:k])
reverse(nums[k:])
}