PTA乙级题目1008(python3)
题目:
一个数组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位以后的整数序列,之间用空格分隔,序列结尾不能有多余空格。
想法:
题目中说不让用额外的数组,也不试过是不是真不让用。
一个想法:保留最后一个数据,然后列表for倒序,每一个向右移(即x+1 = x)。
python3
# 输入第一行的两个数据,可以限制一下读取个数,一般不用
list1 = list(map(int, input().strip().split()))
# 第二行数据,限制读取个数为上一行第一个数字
list2 = [i for i in input().strip().split()[:list1[0]]]
# 右移
for i in range(list1[1]):
# 保留最后一个
a = list2[-1]
# 剩下的向右移
for z in range(len(list2) - 2, -1, -1):
list2[z + 1] = list2[z]
# 最后一个移到第一个
list2[0] = a
# 输出结果,别忘了最后一个没空格
for i in range(len(list2) - 1):
print(list2[i], end=" ")
print(list2[-1])
总结
写完后总以为会超时,结果没有,很惊喜!!!