python解决旋转数组问题

本文讨论了如何在空间复杂度为O(1)的条件下解决数组旋转问题,首先介绍了使用双循环的传统方法,然后提出通过逆序操作降低时间复杂度的优化方案。作者给出了Python代码实现,展示了从优化角度看待数组旋转问题的新思路。
摘要由CSDN通过智能技术生成

如果在限制空间复杂度的基础上,解决一个关于数组旋转的问题,如果说正常不限制空间复杂度的话,是一个较为简单的问题,但是在针对算法进行优化上,把空间复杂度控制在O(1),还是值得考虑深究的。

对于给定一个列表nums,将列表中元素依次向右旋转k位,保证给定的k是非负数,求旋转后的数组。例如:

添加图片注释,不超过 140 字(可选)

添加图片注释,不超过 140 字(可选)

添加图片注释,不超过 140 字(可选)

对于该问题,最直接的想法就是根据给定的k位,将末尾元素逐个向前移动,并且将该元素之前的所有元素均向后移动一位,主要是利用双循环来实现。

    def fuc(nums,k):
        for i in range(k):
            temp= nums[-1]
            for j in range(len(nums)-2,-1,-1):
                nums[j+1] = nums[j]
            nums[0]=temp
        return nums

这种实现的方法时间复杂度主要是来自于逐个移动元素的双重循环,每个元素都被移动了k次,空间复杂度是满足条件的,但是时间复杂度略高,所以考虑优化时间复杂度。

旋转数组k次其实就是将k个尾部元素移动到了头部,同时将剩余元素向后移动,那么将原始数组逆序,同时再对前k个元素逆序,然后剩余的元素逆序,即可得到旋转k次以后得数组了,这种方式相当于对n个元素,每个元素被反转了2次,时间复杂度降低了,空间复杂度仍然是符合条件的。

代码实现如下:

class Solution:
    def func(self, nums, k):
        def reverse(nums,start,end):
            while start<end:
                tmp=nums[start]
                nums[start]=nums[end]
                nums[end]=tmp
                start+=1
                end-=1
        k%=len(nums)
        reverse(nums,0,len(nums)-1)
        reverse(nums,0,k-1)
        reverse(nums,k,len(nums)-1)
        return nums
    

  • 16
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值