leetcode189---旋转数组

leetcode189—旋转数组


一、题目描述

给定一个数组,将数组中的元素向右移动 k k k个位置,其中 k k k 是非负数。

进阶:

  • 尽可能想出更多的解决方案,至少有三种不同的方法可以解决这个问题。
  • 你可以使用空间复杂度为 O ( 1 ) O(1) O(1) 的 原地 算法解决这个问题吗?

示例 1:
输入: nums = [1,2,3,4,5,6,7], k = 3
输出: [5,6,7,1,2,3,4]
解释:
向右旋转 1 步: [7,1,2,3,4,5,6]
向右旋转 2 步: [6,7,1,2,3,4,5]
向右旋转 3 步: [5,6,7,1,2,3,4]

作者:力扣 (LeetCode)
链接:https://leetcode-cn.com/leetbook/read/top-interview-questions-easy/x2skh7/
来源:力扣(LeetCode)


二、方法

要考虑k大于序列长度的情况,所以k对于length要进行取余操作。
即 k = k mod length, k = k%n 。


1.反转数组法

一个数组向右平移k个单位,从结果进行分析:

  1. 前面的元素会向后平移k个单位。
  2. 后面的k个元素会移动到数组的前半部分。
  3. 整个序列保持原有顺序。

因此考虑通过反转数组来进行实现:

  1. 将整个数组进行反转,使得后面的元素移动到前半部分。
  2. 对移动后数组的前k个元素进行反转,使得当前数组满足后k个元素在数组前半部分,且保持原有顺序。
  3. 对数组后面剩余的n-k个元素进行反转,使得顺序恢复正常,并且由于后面k个元素插入了数组顶端,所以相对于原数组完成了向右平移k个单位。
    leetcode189---旋转数组_第1张图片

代码如下:

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.reverse()
    	self.reverseL(nums, 0, k-1)
   		self

你可能感兴趣的:(数组,leetcode)