给定一个数组,将数组中的元素向右移动 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]
说明:
- 尽可能想出更多的解决方案,至少有三种不同的方法可以解决这个问题。
- 要求使用空间复杂度为 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.
"""
# k = k%len(nums)
nums = nums[-k:]+nums[:k+1]
结果输出的还是[1,2,3,4,5,6,7]
但我自己尝试在pycharm里运行输出的结果是正确的,为什么呢?
后来改成nums[:] = nums[-k:] + nums[:k]就正确了,为什么呢?
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.
"""
k = k%len(nums)
nums[:] = nums[-k:] + nums[:k+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.
"""
k = k%len(nums)
if len(nums) == 1 or k ==0 :
nums=nums
# elif len(nums) == 2:
# if k%2==1:
# nums.reverse()
else:
nums[:] = nums[-k:] + nums[:len(nums)-k]
终于成功了