PAT乙级练习题_1008“数组元素循环右移问题”_python解题

原题

一个数组A中存有N（>0）个整数，在不允许使用另外数组的前提下，将每个整数循环向右移M（≥0）个位置，即将A中的数据由( $A_0​​ A_1⋯A_{N−1}$）变换为（ $A_{N−M}⋯A_{​N−1}A_0A_1⋯A_{N−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

my answer

思路

不真正地去移动数组，而是去计算移动数组之后的“第一个数”在原数组中的位置index_start，从index_start开始依次输出数组中对应位置的元素，到数组末尾就回到原数组的首位，直到重新回到index_start时结束。

在计算index_start时，非常简单，index_start = N - (N % M），之后只需要每次加一并模N即可得到index_start位置之后元素的位置（(index_start + i) % N， 其中i是循环的次数）

input_str = input() # 输入数组长度，以及移动的长度
input_numbers = input() # 输入数组
input_str_list = input_str.split()
# 获取M和N
N = int(input_str_list[0]) 
M = int(input_str_list[1])
input_number_list = input_numbers.split()
m = M % N # m是数原数组中第一个数移动之后的位置
index_start = N-m
for i in range(N):
    if i < N-1:
        print(input_number_list[(index_start+i)%N], end=' ')
    else:
        print(input_number_list[(index_start+i)%N])