面试经典(6/150)轮转数组
给定一个整数数组 nums,将数组中的元素向右轮转 k 个位置,其中 k 是非负数。
以下为自己的思路,我不明白最终的返回值为什么有误,好像是题目里要求原地解决问题,而我创建了新数组。
class Solution:
def rotate(self, nums: List[int], k: int) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
#当前数组元素个数-当前索引
#若差值大于k,则i+k
#若差值小于k,则差值+k
#错误思路,赋值以后原来的值不存在了
#为解决以上错误,创建了一个新数组,但是最终返回结果有误
n= len(nums)
new_list = [0]*n
for i in range(len(nums)):
if n-i > k:
new_list[i+k] = nums[i]
print(new_list)
if n-i <= k:
new_list[i-n+k] = nums[i]
print(new_list)
return new_list
正确解法:
直接切片解决,需要注意两点: 1、k需要预先除列表长度取余,因为移动n次相当于没动 2、切片的用法:复制列表时,需要用代码中所示的用法(注意左边nums括号中的冒号,创建了一个新的切片)将旧列表的值赋给新列表,如果直接 list_1 = list_2,两个变量只是指向同一个列表,并非创建了一个新的。
class Solution:
def rotate(self, nums: List[int], k: int) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
n = len(nums)
k = k%n
nums[:] = nums[n-k:]+nums[:n-k]
参考链接:https://leetcode.cn/problems/rotate-array/solutions/1087927/python3-qie-pian-jie-jue-by-westcott-36jh/?envType=study-plan-v2&envId=top-interview-150