旋转数组

要求:

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

示例:

(1)
输入: [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]

(2)
输入: [-1,-100,3,99] 和 k = 2
输出: [3,99,-1,-100]
解释:
向右旋转 1 步: [99,-1,-100,3]
向右旋转 2 步: [3,99,-1,-100]

分析:

将前k个元素移动到数组的后半部分,将后(n-k)个元素移动到数组的前半部分。

代码:
1
2
3
4
5
6
7
8
9
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.
"""
n = len(nums) - k
nums[:] = nums[n:] + nums[:n]
拓展:

1、这是一个直观的算法操作,以切片(分块)的方式操作数组,比较符合人的思维,但是要注重观察示例特征,否则容易陷入单个元素移动的怪圈。

2、nums[:] = 和 nums = 的区别:

1
2
3
4
5
6
7
8
9
10
11
>>> a = list(range(10))
>>> b = a
>>> a[:] = [0, 0, 0] # changes what a and b both refer to
>>> b
[0, 0, 0]

>>> a = list(range(10))
>>> b = a
>>> a = [0, 0, 0] # a now refers to a different list than b
>>> b
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

旋转数组
https://xiepeng21.cn/posts/fb62c7d3/
作者
Peng Xie
发布于
2018年7月17日
许可协议