【LeetCode每日一题】189. 轮转数组

189. 轮转数组

题目

给定一个整数数组 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]

提示：
1 <= nums.length <= 105
-231 <= nums[i] <= 231 - 1
0 <= k <= 105

进阶：
尽可能想出更多的解决方案，至少有 三种 不同的方法可以解决这个问题。
你可以使用空间复杂度为 O(1) 的 原地 算法解决这个问题吗？

总体思路

1. 额外数组法
   新建一个同长度数组 newnums，把老数组第 i 个元素放到新数组的 (i+k) % n 位置（这里 n=len(nums)）。最后再整体拷贝回原数组。

2. 模拟法（每次旋转一步，重复 k 次）：超时，不可行。
   取出数组最后一个元素 n。
   把所有元素整体右移一位。
   把最后一个元素 n 放到最前面。
   重复上面的过程 k 次。
   时间复杂度：O(n*k)

3. 我们要把数组向右移动 k 步，可以通过 三次反转 来实现：
   整体反转数组

   原数组:  [1, 2, 3, 4, 5, 6, 7], k=3
   翻转后: [7, 6, 5, 4, 3, 2, 1]
   

   反转前 k 个元素
   因为这 k 个元素本该出现在前面，所以单独再反转回来：

   [7, 6, 5] -> [5, 6, 7]
   数组变成: [5, 6, 7, 4, 3, 2, 1]
   

   反转后 n-k 个元素
   把剩下的部分反转：

   [4, 3, 2, 1] -> [1, 2, 3, 4]
   数组最终:   [5, 6, 7, 1, 2, 3, 4]
   

   时间复杂度 O(n)，空间复杂度 O(1)

代码

（按照总体思路的顺序）
golang

func rotate(nums []int, k int)  {
    newnums := make([]int,len(nums))
    for i, v:= range nums {
        newnums[(i+k)%len(nums)]=v
    }
    copy(nums,newnums)
}

// 不可行，超时
func rotate(nums []int, k int) {
    s := len(nums)
    for j := 0; j < k; j++ {
        n := nums[s-1]            // 取最后一个
        for i := s-1; i > 0; i-- { // 注意这里是 s-1
            nums[i] = nums[i-1]   // 向右移一位
        }
        nums[0] = n               // 放到最前面
    }
}

package main
import "fmt"

// 反转数组中 [start, end] 范围的元素
func reverse(nums []int, start, end int) {
    for start < end {
        nums[start], nums[end] = nums[end], nums[start]
        start++
        end--
    }
}

func rotate(nums []int, k int) {
    n := len(nums)
    k = k % n // 防止 k 大于数组长度

    // 1. 整体反转
    reverse(nums, 0, n-1)
    // 2. 反转前 k 个
    reverse(nums, 0, k-1)
    // 3. 反转剩下的
    reverse(nums, k, n-1)
}

func main() {
    nums := []int{1,2,3,4,5,6,7}
    rotate(nums, 3)
    fmt.Println(nums) // [5,6,7,1,2,3,4]
}

知识

数组旋转的最佳方法：

三次反转法 (空间 O(1)，时间 O(n)) 🔥 最优解

先整体反转：reverse(nums, 0, n-1)

再反转前 k 个：reverse(nums, 0, k-1)

最后反转剩余部分：reverse(nums, k, n-1)

👉 这样可以在原地完成旋转。

数组复制：可以使用 copy(dest, src)，或直接索引赋值。