给定一个数组,将数组中的元素向右移动 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
-
-
class Solution:
def rotate(self, nums: List[int], k: int) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
# #根据官方方法一改编的python版本
# #先在nums1中填充好数量与nums中相同的一个字符,防止nums1[(i + k) % len(nums)]越界,最后再删除这些东西
# nums1 = ['*'] * len(nums)
# for i in range(len(nums)):
# nums1[(i + k) % len(nums)] = nums[i]
# nums.clear()
# for k in nums1:
# if k == '*':
# nums1.remove(k)
# for j in nums1:
# nums.append(j)
# #方法二 翻转列表三次
# k = k % len(nums)
# nums.reverse()
# nums[:k] = reversed(nums[:k])
# nums[k:] = reversed(nums[k:])
#以下方法不行
k = k % len(nums)
nums = list(reversed(nums))
nums[:k] = list(reversed(nums[:k]))
nums[k:] = list(reversed(nums[k:]))
#另一种方法
k %= len(nums)
nums[:] = nums[-k:] + nums[:-k]
最后一种方法不行的原因:
思路没问题,但是在Leetcode上是无法通过的,这是因为nums = list(reversed(nums))
这段代码对nums进行了赋值操作,reversed()函数
会返回一个generator
,list()函数
将其进行实例化。新的nums指向list(reversed(nums))
所生成的新列表内存地址,而不是在原nums基础上进行修改。也就是说,利用库函数进行反转动用了了新的内存地址,在这道题的要求下是无法通过的。
具体单步调试时,加入id(nums)
查看内存地址:
nums = [1,2,3,4,5,6,7]
print(id(nums))
nums = list(reversed(nums))
print(id(nums))
print(nums)
结果如下:
2728201677960
2728197476296
[7, 6, 5, 4, 3, 2, 1]
可以看到nums确实被重新指向了新的内存地址。