1.题目描述
给定一个数组,将数组中的元素向右移动 k 个位置,其中 k 是非负数。
示例 1:
输入: [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:
输入: [-1,-100,3,99] 和 k = 2
输出: [3,99,-1,-100]
解释:
向右旋转 1 步: [99,-1,-100,3]
向右旋转 2 步: [3,99,-1,-100]
说明:
尽可能想出更多的解决方案,至少有三种不同的方法可以解决这个问题。
要求使用空间复杂度为 O(1) 的 原地 算法。
2.解题思路
题干说要原地算法,即不能创建新的列表,占用额外内存。
解法一:暴力解法
从nums尾部删除的元素插入到nums头部。
解法二:列表拼接
把后面部分和前面部分拼接起来。
解法三:三次反转
先整体反转,再分别对前后部分反转。
总结:暴力解法最慢,列表拼接最快。
3.程序代码
解法一:暴力解法
class Solution:
def rotate(self, nums: List[int], k: int) -> None:
for i in range(k):
nums.insert(0,nums.pop())
时间:140ms;空间:14MB
解法二:
class Solution:
def rotate(self, nums: List[int], k: int) -> None:
n=k%len(nums)
if len(nums)>1 and n>0:
nums[:]=nums[-n:]+nums[:-n]
时间:36ms;空间:13.9MB
解法三:
class Solution:
def rotate(self, nums: List[int], k: int) -> None:
n=k%len(nums)
nums[:]=nums[::-1]
nums[:n]=nums[:n][::-1]
nums[n:]=nums[n:][::-1]
时间:52ms;空间:13.9MB