文巾解题 189. 旋转数组

Posted 2021-09-01 刘文巾

1 题目描述

2 解题思路

2.1 轮转赋值

建立一个辅助数组，辅助数组第(i+k)%len(nums)位的值等于原来的nums数组的第i位的值

class Solution(object):
    def rotate(self, nums, k):
        l=len(nums)
        ret=[0]*l
        for i in range(l):
            ret[(i+k)%l]=nums[i]
        for i in range(l):
            nums[i]=ret[i]

2.2 切片

-k%len(nums)到末尾的值移动到最前面，首位到-k%len(nums)的值移动到最后面

class Solution(object):
    def rotate(self, nums, k):
        l=len(nums)
        ret=nums[-k%l:]+nums[:-k%l]
        
        for i in range(l):
            nums[i]=ret[i]

2.3 翻转

先将整个数组翻转，翻转后的数组的前k位翻转，再将后面的数组翻转

class Solution(object):
    def rotate(self, nums, k):
        l=len(nums)
        nums.reverse()
        nums
        tmp1=list(reversed(nums[:k%l]))
        tmp2=list(reversed(nums[k%l:]))
        lt1=len(tmp1)
        lt2=len(tmp2)
        for i in range(lt1):
            nums[i]=tmp1[i]
        for i in range(lt2):
            nums[i+lt1]=tmp2[i]
        nums