给定一个数组,将数组中的元素向右移动 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) 的原地算法。
#法一:直接双重循环依次移动每一个元素,时间复杂度:O(kn),空间复杂度:O(1)
class Solution:
#法一:直接双重循环依次移动每一个元素,时间复杂度:O(kn),空间复杂度:O(1)
def rotate(self, nums: 'List[int]', k: 'int') -> 'None':
"""
Do not return anything, modify nums in-place instead.
"""
length = len(nums)
#先简化移动次数K
k %= length
for i in range(k):
temp = nums[length-1]
for j in range(length-1,0,-1):
nums[j] = nums[j - 1]
nums[0] = temp
#print(nums)
这是最容易想到的方法,但是像这种双重循环的算法,Python的运行速度会超时,java、C/C++可以AC
# 法二:翻转: 时间复杂度:O(n), 空间复杂度:O(1)
def rotate2(self, nums: 'List[int]', k: 'int') -> 'None':
length = len(nums)
k %= length
self.reverse(nums, 0, length - 1)
self.reverse(nums, 0, k - 1)
self.reverse(nums, k, length - 1)
#print(nums)
def reverse(self,nums: 'List[int]', start: 'int',end: 'int'):
while (start < end):
nums[start], nums[end] = nums[end],nums[start]
start += 1
end -= 1
这种方法是大家普遍使用的方法,不太好理解,但是在稿纸上模拟一遍就懂了!
人生苦短,我用Python,可Python贼慢,我心很烦!所以想要用好Python,还是得加强算法优化能力!