原题目:https://pintia.cn/problem-sets/17/problems/262
一个数组A中存有N(>0)个整数,在不允许使用另外数组的前提下,将每个整数循环向右移M(≥0)个位置,即将A中的数据由(A0A1⋯AN−1)变换为(AN−M⋯AN−1A0A1⋯AN−M−1)(最后M个数循环移至最前面的M个位置)。如果需要考虑程序移动数据的次数尽量少,要如何设计移动的方法?
输入格式:
每个输入包含一个测试用例,第1行输入N(1≤N≤100)和M(≥0);第2行输入N个整数,之间用空格分隔。
输出格式:
在一行中输出循环右移M位以后的整数序列,之间用空格分隔,序列结尾不能有多余空格。
输入样例:
6 2
1 2 3 4 5 6
输出样例:
5 6 1 2 3 4
采用切片的方法,代码如下。
代码:
def move(test,n,m):
test = test[:n-m]+test[-1:-1-m:-1]
test = test[-m-1:-n-1:-1]+test[n-m:]
test = test[::-1]
return test
N,M = input().split(' ')
N = int(N)
M = int(M)
TEST = input().split(' ')
if N >= M:
TEST = move(TEST,N,M)
print(' '.join(TEST))
else:
if M % N == 0:
print(' '.join(TEST))
else:
k = M % N
TEST = move(TEST,N,k)
print(' '.join(TEST))
提交后,在M大于N情况时,返回一个 非零返回 的错误。不知道如何解决 。