题目:
给定一个数组,将数组中的元素向右移动 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]
思路:
这道题我一开始的想法是 建立一个空数组data 然后把原数组要移动的值放进data里 最后在整合
但我一直纠结的问题是 我应该怎么挪这个数 然后就想栈啊队列啊 越想越复杂
最后决定试试数学方法(万物皆可数学
用取余的方法,直接替换掉对应位置的数,即data[(i + k) % len(nums)] = nums[i]
代码:
class Solution:
def rotate(self, nums: List[int], k: int) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
data = [0 for i in range(len(nums))]
for i in range(len(nums)):
data[(i + k) % len(nums)] = nums[i]
for i in range(len(nums)):
nums[i] = data[i]
return nums
但是这么做感觉有点脱裤子放屁的意思,于是再次思考是不是可以直接给data赋值nums,然后操作nums求得结果
改进代码:(这部分其实是有问题的,后面会解释)
class Solution:
def rotate(self, nums: List[int], k: int) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
data = nums
for i in range(len(nums)):
nums[(i + k) % len(nums)] = data[i]
return nums
这么做理论上是没问题的,但run之后的结果是这样的
造成奇怪结果的原因是,data = nums这句话,虽然这样确实是吧nums的值放到data里面了,但是这其实不是赋值操作,这只是一个引用操作,相当于给data传的是nums的首地址,这样导致的问题就是data的值是会根据后边操作变化的,因此最后的结果变得奇怪,所以需要修改一下这句话,变成data = nums.copy(),这样就能保证data的值不会变化。
完整代码:(这次正确了)
class Solution:
def rotate(self, nums: List[int], k: int) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
data = nums.copy()
for i in range(len(nums)):
nums[(i + k) % len(nums)] = data[i]
return nums
题目来自LeetCode第189题