【leetcode】（初级算法-数组）旋转数组【Python】

旋转数组

给定一个数组，将数组中的元素向右移动 k 个位置，其中 k 是非负数。

说明

尽可能想出更多的解决方案，至少有三种不同的方法可以解决这个问题。
要求使用空间复杂度为 O(1) 的 原地算法。

示例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]

方法一：使用临时空间（不符合题目要求）

nums = [1,2,3,4,5,6,7]
k = 3
# 使用临时空间
def rotate(nums, k):
    l = len(nums)
    k %= l
    s2 = [None]*l
    for i in range(l):
        if i < k:
            s2[i] = nums[i+k+1]
        else:
            s2[i] = nums[i-k]
    return s2
print(rotate(nums,k))

方法二：逐个向前移动（测试时会发生超时）

# 逐个向前移动
def rotate(nums, k):
    l = len(nums)
    k %= l
    for i in range(k):
        j = l-1
        temp = nums[l-1]
        while j > 0:
            nums[j] = nums[j - 1]
            j -= 1
        nums[0] = temp
    return nums
print(rotate(nums,k))

方法三：整个区域向前移动

# 整个区域移动
def rotate(nums, k):
    l = len(nums)
    k %= l
    rotatePart = nums[len(nums)-k:]
    nums[k:] = nums[:len(nums)-k]
    nums[:k] = rotatePart
    return nums
print(rotate(nums,k))

方法四：切片方法

def rotate(nums, k):
    l = len(nums)
    k %= l
    nums[:] = nums[-k:] + nums[:-k]
    return nums
print(rotate(nums,k))