轮转数组的两种方法
1.思路
第一种方法对应的思路是像题目中示例给出的解释一样,用循环来做,每一次循环将后面的一个元素移到前面来。第二种方法的思路是直接使用切片来做。
2.解题方法
需要注意k值和nums长度的关系的影响,如果k>=len(nums),需要有单独的逻辑来处理,而这个逻辑也适用于k<len(nums)的情况。
3.复杂度
方法二,时间复杂度O(n),空间复杂度O(1)
4.Code
#方法一
class Solution:
def rotate(self, nums: List[int], k: int) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
if k==0:
return
else:
index=0
rotate_num=k%len(nums)
while rotate_num:
slice=nums[-rotate_num]
nums.pop(-rotate_num)
nums.insert(index,slice)
index+=1
rotate_num-=1
#方法二
class Solution:
def rotate(self, nums: List[int], k: int) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
length=len(nums)
if k==0 or k%length==0:
return
else:
slice=nums[-k%length:]
nums[:]=slice+nums[0:len(nums)-k%length]