#python #leetcode189轮转数组

题目

#python #leetcode189轮转数组_第1张图片

算法1

  1. 创建新列表,存储循环右移之后的新序列

原序列下标i右移k之后的新下标为(i+k)%n,因此得到的新序列ans[new_index[i]] = nums[i]

  1. 将新列表拷贝到旧列表中

代码1

class Solution:
    def rotate(self, nums: List[int], k: int) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        n = len(nums)
        index = list(range(n))

        new_index = [(i + k) % n for i in index]
        ans = [0]*n
        
        for i in range(n):
            ans[new_index[i]] = nums[i]
   
        nums[:] = copy.deepcopy(ans)

算法2

容易发现,循环右移的规律就是把原始数组最右边k个元素移动到最前面作为新数组前k个元素,将前n-k个数组元素后移k位作为新数组最后n-k个元素。

故使用python切片操作

a[:]=a[n-k:]+a[0:n-k]
class Solution:
    def rotate(self, nums: List[int], k: int) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        n = len(nums)
        k = k % n
        nums[:] = nums[n - k:] + nums[:n - k]

作者:westcott
链接:https://leetcode-cn.com/problems/rotate-array/solution/python3-qie-pian-jie-jue-by-westcott-36jh/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

cheat sheet

  1. 列表不能直接赋值,要用切片
  2. 切片是非常灵活的,python是非常灵活的
  3. 多做笔记

你可能感兴趣的:(leetcode,python,算法)