这里采用的是一种比较取巧的方法,和考研书王道上的方法相同,是利用镜像的原理一次性把数组翻转到对应位置。
class Solution:
def reverse(self,nums: List[int], a: int, b:int) -> None:
while(a < b):
temp = nums[b]
nums[b] = nums[a]
nums[a] = temp
a += 1
b -= 1
def rotate(self, nums: List[int], k: int) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
l =len(nums)
k = k % l
l = l-1
self.reverse(nums,0,l-k)
self.reverse(nums,l-k+1,l)
self.reverse(nums,0,l)