python - 阵列左旋转

Question

我有一个问题，我需要将数组向左旋转 k 次。

即如果 k = 2, [1, 2, 3, 4, 5] 。 -> [3, 4, 5, 1, 2]

所以，我的代码是:

def array_left_rotation(a, n, k):
    for i in range(n):
        t = a[i]
        a[i] = a[(i+n-1+k)%n]
        a[(i+n-1+k)%n] = t

    return a

其中 n = 数组的长度。

我认为问题是映射问题，a[0] -> a[n-1] if k = 1。

我的解决方案有什么问题？

Answer 1

另一种借助索引来实现这一点的方法如下所示。

def rotate(l, n):
    return l[n:] + l[:n]

print(rotate([1, 2, 3, 4, 5], 2))

#output : [3, 4, 5, 1, 2]

只有当 n 超出范围 [-len(l), len(l)] 时，才会返回原始列表。要使其适用于 n 的所有值，请使用:

def rotate(l, n):
  return l[-n % len(l):] + l[:-n % len(l)]