【浙江大学PAT真题练习乙级】1008 数组元素循环右移问题 (20分)真题解析

题目描述

  一个数组$A$中存有$N$(>0)个整数，在不允许使用另外数组的前提下，将每个整数循环向右移$M( \geq 0)$个位置，即将$A$中的数据由($A_{0}A_{1},\cdots,A_{N-1}$)变换为($A_{N-M}\cdots A_{N-1}A_{0}A_{1} \cdots ,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

思路解析

  一开始我的想法就是将M-N个数据(也就是前面的数据存储起来，因为往后挪的话，后面的数据会跑到前面先输出，比如样例中的5，6，所以1，2，3，4就需要先存储起来)，也就是创建一个M-N的数组，用于最后输出。

  还要考虑$M\geq N$的情况，这里采用$M=M\%N$的方式解决。如果是$N$的整数倍的话就全部要存入数组M_num中，否则的话就只需要存储前面需要挪到后面的M-N个数。

  最后一个是无空格的，所以在输出数组的最后一个元素时要分开判断一下。

C++实现

#include<stdio.h>

int main(){
    int N,M;
    scanf("%d %d", &N, &M);
    M = M%N;
    int input_num;
    
    int M_num[N-M];
    for(int i=0; i<N; i++){
        scanf("%d", &input_num);
        if(i<(N-M)){
            M_num[i] = input_num;
        }
        else{
            printf("%d ", input_num);
        }
    }
    for(int j=0; j<(N-M);j++){
        
        
        if(j==N-M-1){
        printf("%d", M_num[j]);
        }
        else{
            printf("%d ",M_num[j]);
        }
    }
    
    
    return 0;
}

Python实现

  ·python·实现的话就可以直接用数组切片操作了。

N_M =  input().split(" ")
input_num = input().split(" ")

N = int(N_M[0])
M = int(N_M[1])

M = M%N
input_num = input_num[N-M:]+input_num[:N-M]

for i in range(N):
    if i == N-1:
        print(input_num[i], end="")
    else:
        print(input_num[i], end=" ")