这个题是个easy题,但是题目说的是用尽量多的解法来解,而且是尽量小的时间空间复杂度。
先看题。
题目:
189. 旋转数组
给定一个数组,将数组中的元素向右移动 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]
好了,用最多的解法来解。共有四种,官方解析。
看python代码
code:
class Solution:
def rotate(self, nums: List[int], k: int) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
# 1. 直接多次循环,一个一个往后移
# 时间复杂度O(n*k),空间复杂度O(1)
# Time Limit Exceeded
n = len(nums)
k = k%n
for i in range(k): # 移动k次
cur = nums[-1]
for j in range(n): # 所有数字都往右移
temp = nums[j]
nums[j] = cur
cur = temp
return nums
def rotate(self, nums: List[int], k: int) -> None:
# 2. 使用额外数组
# 时间复杂度O(n),空间复杂度O(n)
# Accepted
n = len(nums)
k = k%n
res1 = []
res2 = []
for i in range(n):
if i < n-k:
res2.append(nums[i])
else:
res1.append(nums[i])
for i, n in enumerate(res1+res2):
nums[i] = n
return nums
def rotate(self, nums: List[int], k: int) -> None:
# 3. 使用环形列表,直接将每个元素放到自己的位置
# 时间复杂度O(n),空间复杂度O(1)
# 太麻烦了,转来转去的,不要用这种方法了
# 建议大概看看就好了
def rotate(self, nums: List[int], k: int) -> None:
# 4. 多次翻转
# 时间复杂度O(n),空间复杂度O(1)
# Accepted
def reverse(num, l, r):
while l<r:
num[l], num[r] = num[r], num[l]
l += 1
r -= 1
n = len(nums)
k = k%n
reverse(nums, 0, n-1)
reverse(nums, 0, k-1)
reverse(nums, k, n-1)
return nums
好了。多看,多想。https://leetcode-cn.com/problems/rotate-array/solution/xuan-zhuan-shu-zu-by-leetcode/