python实现数组平移K位问题

def move(ls: list, offset):
    """
    元素原索引+位移数（正为右移，负为左移）之和求关于数组长度（数组的模）的余数，即为位移后的元素索引。
    再对新索引升序排序，去除索引，即为位移后的新数组
    :param ls:
    :param offset:
    :return:
    """
    mod = len(ls)
    ids = [[(item[0]+offset)%mod, item[1]] for item in enumerate(ls)]
    ids.sort(key=lambda item: item[0])
    return [item[1] for item in ids]

def move2(ls: list, offset):
    """
    分段反转，以位移数（对模求余）为界，分别反转两个子数组，再整体反转
    :param ls:
    :param offset:
    :return:
    """
    mod = len(ls)
    offset = offset % mod
    tail = list(reversed(ls[mod-offset:]))
    head = list(reversed(ls[:mod-offset]))
    return list(reversed(head+tail))

if __name__ == '__main__':
    nums = [8, 9, 10, 11]
    print(move(nums, 1))
    print(move2(nums, 1))

    print(move(nums, -1))
    print(move2(nums, -1))
    """
	[11, 8, 9, 10]
	[11, 8, 9, 10]
	[9, 10, 11, 8]
	[9, 10, 11, 8]
	"""

#方法一
# -*- 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:
            arr[i] = arr[i-1] #所有元素后移
            i -= 1
        arr[0] = tmp #第一个元素为初始最后一个元素的值
        k -= 1
 
if __name__ == "__main__":
    k = 4
    arr = ['a','b','c','d','1','2','3','4']
    rightShift(arr,k)
    i = 0
    while i < len(arr):
        print(arr[i],end="")
        i += 1

#方法二
def reverse(arr,start,end):
    while start<end:
        temp = arr[start]
        arr[start] = arr[end]
        arr[end] = temp
        start += 1
        end -= 1
 
def rightShift(arr,k):
    if arr == None:
        print("参数不合法！")
        return
    lens = len(arr)
    k %= lens
    reverse(arr,0,lens-k-1)
    reverse(arr,lens-k,lens-1)
    reverse(arr,0,lens-1)
 
if __name__ == "__main__":
    k = 4
    arr = ['a','b','c','d','1','2','3','4']
    rightShift(arr,k)
    i = 0
    while i < len(arr):
        print(arr[i],end="")
        i += 1