我正在尝试解决HackerRank中的一个问题,当我提交我的解决方案时,我收到一条错误,指出“由于超时而终止” .
请检查代码并建议我如何优化 .
Question: 对n个数组的数组进行左旋转操作会将每个数组的元素1单元向左移动 . 例如,如果在数组[1,2,3,4,5]上执行了2次左旋转,则数组将变为[3,4,5,1,2] .
给定n个整数和数字d的数组,在数组上执行左旋转 . 然后将更新的数组打印为单行空格分隔的整数 .
Input Format
第一行包含两个以空格分隔的整数,表示n(整数)和d(必须执行的左旋转数)的相应值 . 第二行包含空格分隔的整数,用于描述数组初始状态的各个元素 .
Output Format
在执行左旋转后,打印一行n个以空格分隔的整数,表示数组的最终状态 .
Sample Input
5 4 1 2 3 4 5
Sample Output
5 1 2 3 4
Explanation
当我们执行d = 4左旋转时
因此,我们将数组的最终状态打印为单行空格分隔值,即5 1 2 3 4 .
My Code:
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=' ')