文章目录
Leetcode 189: 轮转数组(python)
注:本人文章中有时有来自于各大佬的经验总结,仅供自身复习及学习,并无商业用途,若有时忘记标注作者及出处还请提醒,敬请见谅。
我的思路
python内置的列表操作函数真的很方便,所以我的想法也就很直观:
假设数组大小为n,当所需要移动的位数 k <= n 时,即新数组的前k位等于原数组的最后k位,剩下的部分同理。
具体的如图所示:
如果k > n,则取 k = k % n,使得k <= n,然后执行上面的过程。
代码如下:
class Solution:
def rotate(self, nums: List[int], k: int) -> None:
n = len(nums)
k = k % n
nums[:k], nums[k:] = nums[n-k:], nums[:n-k]