Python1334 · 旋转数组

Python1334 · 旋转数组

    def rotate(self, nums, k):
        # Write your code here
        if k<=len(nums) and len(nums)!=1:
            ListA=nums[-k::]
            ListB=nums[0:len(nums)-k]
            ListA.extend(ListB)
            return ListA
        elif len(nums)==1:
            return nums
        else:
            k%=len(nums)
            ListA=nums[-k::]
            ListB=nums[0:len(nums)-k]
            ListA.extend(ListB)
            return ListA