面试经典(6/150)轮转数组

面试经典(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

面试经典(6/150)轮转数组_第1张图片
正确解法:
直接切片解决,需要注意两点: 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

你可能感兴趣的:(python,笔记,学习)