如果在限制空间复杂度的基础上,解决一个关于数组旋转的问题,如果说正常不限制空间复杂度的话,是一个较为简单的问题,但是在针对算法进行优化上,把空间复杂度控制在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