Python程序因超时终止-HackerRank

Question

我正在尝试解决一个来自HackerRank的问题，当我提交我的解决方案时，我得到一个错误，说明“由于超时而终止”。

问题：对n个大小的数组的左旋转操作将数组的每个元素向左移动1个单位。例如，如果对数组[1,2,3,4,5]执行两次左旋转，那么该数组将变为[3,4,5,1,2]。

给定一个由n个整数和一个数字d组成的数组，对该数组执行d个左旋转。然后将更新后的数组打印为单行以空格分隔的整数。

输入格式

样本输入

5 4 1 2 3 4 5

样本输出

因此，我们将数组的最终状态打印为一行空格分隔的值，即5 1 2 3 4。

我的代码：

def array_left_rotation(ar, n, k):
    for j in range(k):
        temp = ar[0]
        for i in range(0,n-1):
            ar[i] = ar[i+1]
        ar[n-1] = temp
    return ar
n, k = map(int, input().strip().split(' '))
a = list(map(int, input().strip().split(' ')))
answer = array_left_rotation(a, n, k);
print(*answer, sep=' ')

Answer 1

在这里，最好的方法是不要自己实际执行操作。在这种情况下，您是手动旋转列表，这是不必要的计算。

因此，首先，分析问题及其对各种输出的行为。让我们分析一下：

假设一个由5个元素组成的数组

array = [1,2,3,4,5]

[3,4,5,1,2]

[3,4,5,1,2]

类似的试验将会看到这种模式的出现。例如，对于k次的循环，它与k%n次相同。

现在我们有了这些，现在开始讨论计算部分。为此，只需使用列表切片来获得循环，如下所示：

#n -> number of elements in array
#k -> number of rotations to be performed
#a -> (list) array

def rotate(a,n,k) :
    rotations = k % n
    new_array = a[rotations:] + a[:rotations]

    return new_array