萌新练习写代码的每日一练:旋转数组

题目:
给定一个数组,将数组中的元素向右移动 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]

思路:
这道题我一开始的想法是 建立一个空数组data 然后把原数组要移动的值放进data里 最后在整合
但我一直纠结的问题是 我应该怎么挪这个数 然后就想栈啊队列啊 越想越复杂
最后决定试试数学方法(万物皆可数学
用取余的方法,直接替换掉对应位置的数,即data[(i + k) % len(nums)] = nums[i]

代码:

class Solution:
    def rotate(self, nums: List[int], k: int) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        data = [0 for i in range(len(nums))]
        for i in range(len(nums)):
            data[(i + k) % len(nums)] = nums[i]
		for i in range(len(nums)):
			nums[i] = data[i]
        return nums

但是这么做感觉有点脱裤子放屁的意思,于是再次思考是不是可以直接给data赋值nums,然后操作nums求得结果

改进代码:(这部分其实是有问题的,后面会解释)

class Solution:
    def rotate(self, nums: List[int], k: int) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        data = nums
        for i in range(len(nums)):
            nums[(i + k) % len(nums)] = data[i]

        return nums

这么做理论上是没问题的,但run之后的结果是这样的
很奇怪的结果
造成奇怪结果的原因是,data = nums这句话,虽然这样确实是吧nums的值放到data里面了,但是这其实不是赋值操作,这只是一个引用操作,相当于给data传的是nums的首地址,这样导致的问题就是data的值是会根据后边操作变化的,因此最后的结果变得奇怪,所以需要修改一下这句话,变成data = nums.copy(),这样就能保证data的值不会变化。

完整代码:(这次正确了)

class Solution:
    def rotate(self, nums: List[int], k: int) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        data = nums.copy()
        for i in range(len(nums)):
            nums[(i + k) % len(nums)] = data[i]

        return nums

题目来自LeetCode第189题

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值