题目描述
把数组向右轮转 k 步,要求尽量原地完成。
nums = [1,2,3,4,5,6,7], k=3
输出 = [5,6,7,1,2,3,4]
思路解析
关键不是背模板,而是看懂为什么这题适合数组 · 原地反转。
1. 先把 k 对 n 取模,避免白转圈:先把 k 对 n 取模,避免白转圈
2. 原数组分成 A=[1,2,3,4] 和 B=[5,6,7]:原数组分成 A=[1,2,3,4] 和 B=[5,6,7]
3. 右转目标就是 B+A:右转目标就是 B+A
4. 整体反转得到 reverse(B)+reverse(A):整体反转得到 reverse(B)+reverse(A)
5. 反转前 k 个,把 reverse(B) 变回 B:反转前 k 个,把 reverse(B) 变回 B
6. 反转后半段,把 reverse(A) 变回 A:反转后半段,把 reverse(A) 变回 A
7. 数组变成 [5,6,7,1,2,3,4]:数组变成 [5,6,7,1,2,3,4]
8. 三次原地反转完成轮转:三次原地反转完成轮转
把这句话记住,下次遇到同类题,就能更快选出方向。
用一个小问题检查自己是不是真的懂了。
参考代码(Python)
Python
1class Solution:
2 def rotate(self, nums, k):
3 n = len(nums)
4 k %= n
5 def rev(l, r):
6 while l < r:
7 nums[l], nums[r] = nums[r], nums[l]
8 l += 1
9 r -= 1
10 rev(0, n - 1)
11 rev(0, k - 1)
12 rev(k, n - 1)复杂度分析
- 时间复杂度:O(n) —— 每个核心状态按算法要求处理固定次数
- 空间复杂度:O(1) —— 只保存必要的辅助结构或递归栈
套路模板
模板不是死背,而是提醒你写代码前先把状态、转移和边界排好。
Python
1# 数组 · 原地反转 通用检查表
2# 1. 定义状态/指针/容器
3# 2. 每轮只做一个清晰动作
4# 3. 更新答案并处理边界易错点
- 错误写法:忘记 k %= n → 正确写法:先对 k 取模(k 可能比数组长度大很多)
- 错误写法:只按样例推代码 → 正确写法:先写清状态含义和边界条件(样例太少,隐藏用例专打边界)
- 错误写法:变量名和动画不一致 → 正确写法:代码变量沿用动画里的核心名字(学习时最怕脑内维护两套概念)
以上文字与上方动画完全同一思路:先看动画建立直觉 → 读文字巩固 → 关掉页面自己默写一遍代码,能写出来才算真的掌握。