189. Rotate Array -- Python

最新推荐文章于 2024-08-01 23:10:37 发布

Chris_zhangrx

最新推荐文章于 2024-08-01 23:10:37 发布

阅读量593

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/Chris_zhangrx/article/details/79060398

leetcode 专栏收录该内容

51 篇文章

订阅专栏

本文介绍了一种将数组元素向右旋转k步的操作，并提供了两种不同的Python实现方案。一种是通过循环插入和删除元素来完成旋转，另一种则是利用切片操作简化代码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

189. Rotate Array

Rotate an array of n elements to the right by k steps.

For example, with n = 7 and k = 3, the array [1,2,3,4,5,6,7] is rotated to [5,6,7,1,2,3,4].

Note:
Try to come up as many solutions as you can, there are at least 3 different ways to solve this problem.

思路：
就是移动 k 位元素到列表的左端。第一次提交没有考虑到 k 的值比列表长度还大的情况，第二版中加了一个 if 语句判断。

代码：

class Solution:
    def rotate(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: void Do not return anything, modify nums in-place instead.
        """
        if k <len(nums):
            for i in range(k):
                nums.insert(i,nums[-k+i])
                del nums[-k+i]
        else:
            self.rotate(nums,k%len(nums))

题目中提到了可以尝试采用多种方法，所以也参考了一下别人的思路，很简洁。
参考代码1：

class Solution:
    def rotate(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: void Do not return anything, modify nums in-place instead.
        """
        k = k % len(nums)
        nums[:] = nums[-k:] + nums[:-k]

复习收获：
使用list存储数据时，按索引访问元素很快，但是插入和删除元素就很慢了，因为list是线性存储，数据量大的时候，插入和删除效率很低。deque是为了高效实现插入和删除操作的双向列表，适合用于队列和栈：

from collections import deque
>>> q = deque(['a', 'b', 'c'])
>>> q.append('x')
>>> q.appendleft('y')
>>> q
deque(['y', 'a', 'b', 'c', 'x'])

上面可得：
deque除了实现list的append()和pop()外，还支持appendleft()和popleft()，这样就可以非常高效地往头部添加或删除元素。