版权声明:转载请注明! https://blog.csdn.net/linfeng886/article/details/81981192
一、题目
给定一个数组,将数组中的元素向右移动 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]
说明:
1.尽可能想出更多的解决方案,至少有三种不同的方法可以解决这个问题。
2.要求使用空间复杂度为 O(1) 的原地算法。
二、解法
解法一
以倒数第
k
个值为分界线,把nums
截成两组再组合。因为k
可能大于nums
的长度(当这两者相等的时候,就相当于nums
没有移动),所以我们取k % len(nums)
,k
和nums
的长度取余,就是最终我们需要移动的位置
代码如下:
if nums:
k = k % len(nums)
nums[:]=nums[-k:]+nums[:-k]
时间:64ms,击败了98%
解法二
先把
nums
最后一位移动到第一位,然后删除最后一位,循环k次。k = k % len(nums)
,取余扫描二维码关注公众号,回复: 2908112 查看本文章
代码如下:
if nums:
k = k % len(nums)
while k > 0:
k -= 1
nums.insert(0, nums[-1])
nums.pop()
时间:172ms,击败了16%
解法三
先把
nums
复制到old_nums
,然后nums
中索引为x
的元素移动k
个位置后,当前索引为x+k
,其值为old_nums[x]
。,所以我们把x+k
处理成(x+k)%len(nums)
,取余操作,减少重复的次数。
代码如下:
if nums:
old_nums = nums[:]
l = len(nums)
for x in range(l):
nums[(x+k) % l] = old_nums[x]
时间:64ms,击败了98%
结语:
欢迎关注我的公众号
疯子的Python笔记