做网站用哪种语言好免费的行情网站app
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
- 时间复杂度：O(n)
- 空间复杂度：O(1)
总结
通过使用三次反转法，我们可以高效地将数组中的元素向右轮转 k 个位置，同时保持原地操作。这种方法既简单又高效，适合大多数场景。希望这篇题解对大家有所帮助，如果有任何问题，欢迎在评论区留言讨论！
关注我，获取更多算法题解和编程技巧！