题目:
难度中等1079收藏分享切换为英文接收动态反馈
给定一个数组,将数组中的元素向右移动 k
个位置,其中 k
是非负数。
进阶:
- 尽可能想出更多的解决方案,至少有三种不同的方法可以解决这个问题。
- 你可以使用空间复杂度为 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]
示例 2:
输入:nums = [-1,-100,3,99], k = 2 输出:[3,99,-1,-100] 解释: 向右旋转 1 步: [99,-1,-100,3] 向右旋转 2 步: [3,99,-1,-100]
提示:
1 <= nums.length <= 2 * 104
-231 <= nums[i] <= 231 - 1
0 <= k <= 105
注意原地的要求
我的一开始的解法是延长数组,用for非常慢
class Solution:
def rotate(self, nums: List[int], k: int) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
for i in range(k):
nums.append(0)
n = len(nums)
for i in range(n-k-1,-1,-1):
nums[i+k] = nums[i]
for j in range(k-1,-1,-1):
nums[j] = nums[n-k+j]
for m in range(k):
nums.pop()
放在这里,仅供鞭尸和自嘲
然后考虑了申请新的数组,其实申请新的变量就行了,做麻烦了,继续鞭尸
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
m = [0]*k
for i in range(k):
m[-i] = nums[-i]
for i in range(n-k):
nums[-i] = nums[-i-k]
for i in range(k):
nums[(-i+k)%n] = m[-i]
接着考虑切片的方法,没有相当拼接,写的还是很菜
根据评论区学了拼接之后:
class Solution:
def rotate(self, nums: List[int], k: int) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
# python方便的是对于list切片进行拼接,直接通过加和完成
K=k%len(nums)
nums[:]=nums[-K:]+nums[:-K]
算是python处理list数组的特点,所以要学这个,然后利用这样的特点去设计
注意%k哟!
最后实现了一下官方给的翻转数组的办法,那个由于输入[1]和0的bug我不得不写if
但是属实不美观,现在没想好怎么优雅地改
class Solution:
def rotate(self, nums: List[int], k: int) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
n = len(nums)
if k%n!=0:
nums.reverse()
nums[k%n-1::-1] = nums[:k%n:] #如果没有if判断的话那么这里就会出错,左边1个元素右边没有
nums[:k%n-1:-1] = nums[k%n::]
翻转算是C++等语言的vector容器中比较容易进行的,所以从工具特性角度去考虑这样的算法有道理的
另外python我试了一下部分翻转还是挺不方便的感觉,这个下标很容易错,reverse又不让传参数(查官方文档)...
小菜鸡逐渐变强的一天...