分类:array
难度:easy
给定一个数组,将数组中的元素向右移动 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]
说明:
尽可能想出更多的解决方案,至少有三种不同的方法可以解决这个问题。
要求使用空间复杂度为 O(1) 的 原地 算法。
题解1(反转法)
我们可以看到往右移动k个单位,原本后k个元素调到开头,前面n-k个元素依次往后推,所以可以将数组全部反转,然后将前k个反转,再将剩下的n-k个元素反转即可
时间复杂度O(n)
空间复杂度O(1)
代码
class Solution(object):
def rotate(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: None Do not return anything, modify nums in-place instead.
"""
# 反转法
k = k%len(nums)
def reverse(nums,left,right):
curl,curr = left,right
while curl<curr:
nums[curl],nums[curr]=nums[curr],nums[curl]
curl+=1
curr-=1
reverse(nums,0,len(nums)-1)
reverse(nums,0,k-1)
reverse(nums,k,len(nums)-1)
题解2(环状替换)
晚上补充一下