给定一个数组,将数组中的元素向右移动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]
说明:
- 尽可能想出更多的解决方案,至少有三种不同的方法可以解决这个问题。
- 要求使用空间复杂度为O(1)的原地算法。
这个感觉很简单,用切片直接写:
class Solution:
def rotate(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: void Do not return anything, modify nums in-place instead.
"""
nums_len = len(nums)
nums[:] = nums[nums_len - k:] + nums[: nums_len - k]
print (nums)