【浙江大學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=" ")