Python算法——对数组进行循环移位

原创

于 2019-08-11 19:01:22 发布 · 5k 阅读

CC 4.0 BY-SA版权

这篇博客探讨了如何对包含N个元素的数组进行循环右移K位的算法，要求时间复杂度为O(N)。文章分析了两种方法：蛮力法和翻转法。蛮力法通过每次元素右移一位，共移动K次，时间复杂度为O(K*N)。而翻转法通过两次翻转数组部分实现，时间复杂度为O(N)，并满足只使用两个附加变量的条件。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

要求：

对含有N个元素的数组循环右移K位，要求时间复杂度为O(N)，且只允许使用两个附加变量。

分析：

方法一：蛮力法

要求将数组元素循环右移K位，只需要每次将数组中元素右移一位，循环K次即可。如原数组为abcd1234，右移4位具体移动过程为abcd1234-->4abcd123-->34abcd12-->1234abcd。

方法二：翻转法

直接上例子，对于数组序列A = [123456]，如何实现循环右移2位功能？将数组A分成两个部分A[0,N-K-1]和A[N-K,N-1]，将这两部分分别翻转，然后放在一起再翻转，具体如下：

①翻转1234:123456-->432156

②翻转56:432156-->432165

③翻转432165:432165-->561234

代码实现：

#方法一
# -*- coding:utf-8 -*-
def rightShift(arr,k):
    if arr == None:
        print("参数不合法！")
        return
    lens = len(arr)
    k %= lens #因为K不一定小于N，有可能大于等于N，当K≥N时，右移K-N与右移K位效果一样
    while k != 0: #右移k位
        tmp = arr[lens-1] #数组最后一个元素放入临时变量中
        i = lens-1
        while i > 0: