LeetcodeHot100:189. 轮转数组

leetcode189. 轮转数组：解题思路与代码实现

问题描述

给定一个整数数组 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 <= 10^5
• -2^31 <= nums[i] <= 2^31 - 1
• 0 <= k <= 10^5

解题思路

方法一：使用额外空间

最直观的方法是使用额外的数组来存储旋转后的结果。具体步骤如下：

1. 创建一个与 nums 相同长度的数组 result。
2. 将 nums 中的元素按照旋转后的位置放入 result 中。
3. 将 result 赋值回 nums。

这种方法的时间复杂度为 O(n)，空间复杂度为 O(n)。

方法二：原地旋转

我们可以通过原地修改数组来实现旋转，而不需要额外的空间。具体步骤如下：

1. 计算 k % n，其中 n 是数组的长度，以确保 k 在 0 到 n-1 之间。
2. 将数组的后 k 个元素存储在一个临时数组 tmp 中。
3. 将数组的前 n-k 个元素向后移动 k 个位置。
4. 将 tmp 中的元素赋值到 nums 的前 k 个位置。

这种方法的时间复杂度为 O(n)，空间复杂度为 O(k)。

方法三：三次反转

还有一种更巧妙的方法，通过三次反转来实现旋转：

1. 反转整个数组。
2. 反转前 k 个元素。
3. 反转剩下的 n-k 个元素。

这种方法的时间复杂度为 O(n)，空间复杂度为 O(1)。

代码实现

方法一：使用额外空间

class Solution:
    def rotate(self, nums: List[int], k: int) -> None:
        n = len(nums)
        k %= n
        result = [0] * n
        for i in range(n):
            result[(i + k) % n] = nums[i]
        nums[:] = result

方法二：原地旋转

class Solution:
    def rotate(self, nums: List[int], k: int) -> None:
        n = len(nums)
        k %= n
        tmp = nums[n - k:].copy()
        for i in range(n - k - 1, -1, -1):
            nums[i + k] = nums[i]
        for i in range(0, k):
            nums[i] = tmp[i]

方法三：三次反转

class Solution:
    def rotate(self, nums: List[int], k: int) -> None:
        n = len(nums)
        k %= n
        self.reverse(nums, 0, n - 1)
        self.reverse(nums, 0, k - 1)
        self.reverse(nums, k, n - 1)
    
    def reverse(self, nums: List[int], start: int, end: int) -> None:
        while start < end:
            nums[start], nums[end] = nums[end], nums[start]
            start += 1
            end -= 1