当前位置：首页 > news >正文

LeetCode 热题 100 189. 轮转数组

news 来源：原创 2025/5/5 6:57:04

LeetCode 热题 100 | 189. 轮转数组

大家好，今天我们来解决一道经典的算法题——轮转数组。这道题在LeetCode上被标记为中等难度，要求我们将数组中的元素向右轮转 k 个位置。下面我将详细讲解解题思路，并附上Python代码实现。

问题描述

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

解题思路

核心思想

三次反转法：
- 首先反转整个数组。
- 然后反转前 k 个元素。
- 最后反转剩下的 n - k 个元素。
原地操作：
- 通过反转操作，避免使用额外的空间。

Python代码实现

def rotate(nums, k):n = len(nums)k = k % n  # 处理k大于数组长度的情况def reverse(start, end):while start < end:nums[start], nums[end] = nums[end], nums[start]start += 1end -= 1reverse(0, n - 1)  # 反转整个数组reverse(0, k - 1)   # 反转前k个元素reverse(k, n - 1)   # 反转剩下的元素# 测试示例
nums1 = [1, 2, 3, 4, 5, 6, 7]
nums2 = [-1, -100, 3, 99]rotate(nums1, 3)
rotate(nums2, 2)print(nums1)  # 输出: [5, 6, 7, 1, 2, 3, 4]
print(nums2)  # 输出: [3, 99, -1, -100]

代码解析

处理 k 大于数组长度的情况：
- 使用 k = k % n 确保 k 在数组长度范围内。
定义反转函数：
- reverse 函数用于反转数组中从 start 到 end 的部分。
三次反转操作：
- 第一次反转整个数组，将后 k 个元素移到前面，但顺序相反。
- 第二次反转前 k 个元素，恢复其原始顺序。
- 第三次反转剩下的 n - k 个元素，恢复其原始顺序。
原地操作：
- 直接在原数组上进行操作，不需要额外的空间。

复杂度分析

时间复杂度：O(n)，其中 n 是数组的长度。我们只需要遍历数组三次。
空间复杂度：O(1)，只使用了常数个额外空间。

示例运行

示例1

输入: nums = [1, 2, 3, 4, 5, 6, 7], k = 3
输出: [5, 6, 7, 1, 2, 3, 4]

示例2

输入: nums = [-1, -100, 3, 99], k = 2
输出: [3, 99, -1, -100]

进阶：其他解法

除了三次反转法，我们还可以使用其他方法解决这个问题：

方法一：使用额外数组

def rotate_extra(nums, k):n = len(nums)k = k % nrotated = nums[-k:] + nums[:-k]nums[:] = rotated

时间复杂度：O(n)
空间复杂度：O(n)

方法二：环状替换

def rotate_cyclic(nums, k):n = len(nums)k = k % ncount = 0start = 0while count < n:current = startprev = nums[start]while True:next_idx = (current + k) % nnums[next_idx], prev = prev, nums[next_idx]current = next_idxcount += 1if current == start:breakstart += 1