如何在 Python 中移动或旋转数组

Python 2周前 14

Python 是一种十分强大的编程语言，提供了丰富的操作数组的方法。其中一个很常见的操作是移动或旋转数组。在这篇文章中，我们将深入探讨如何在 Python 中对数组进行移动或旋转，并提供相关的注意事项。

移动数组

移动数组即是将数组中的每个元素向左或向右移动若干个位置。在 Python 中，可以通过切片和列表的拼接来实现移动数组的操作。

例如，如果想要将列表 nums 中的元素向左移动 k 个位置，可以使用以下代码：

k = 3
nums = [1, 2, 3, 4, 5]

nums = nums[k:] + nums[:k]

print(nums)

输出结果为：

[4, 5, 1, 2, 3]

代码中，我们先定义了移动的位置 k 和要移动的列表 nums。接着，我们将 nums 切片成两个部分，先取后面的 k 个元素，再取前面的剩余元素，并将其拼接在一起。最后，我们打印出移动后的列表 nums。

旋转数组

旋转数组是将整个数组旋转若干个位置，可以采用循环移位的方法实现。我们首先将整个数组翻转，再将前 k 个元素翻转，将后面的元素翻转。例如，如果要把数组 nums 向右旋转 k 个位置，可以使用以下代码：

k = 3
nums = [1, 2, 3, 4, 5]

def reverse(start, end, nums):
    while start < end:
        nums[start], nums[end] = nums[end], nums[start]
        start += 1
        end -= 1

n = len(nums)
k %= n

reverse(0, n-1, nums)
reverse(0, k-1, nums)
reverse(k, n-1, nums)

print(nums)