《PTA——拼题A》之第1008题

题目介绍

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

源代码

#include<cstdio>
const int MAXSIZE = 101;

void move(int A[], int N, int M){
  for(; M < N - 1; ++M){
    A[N-M] = A[N-M-1];
  }
  A[1] = A[0];
}

int main(){
  int N, M, A[MAXSIZE]={0};  //A[0]用来存储需要移动的元素
  scanf("%d %d", &N, &M);
  for(int i = 1; i <= N; ++i){
    scanf("%d", &A[i]);
  }
  M = M % N;  //  确保M比N要小
  if(N == M){
    for(int i = N; i >= 1; --i){
    if( i == 1 )
      printf("%d", A[i]);
    else printf("%d ", A[i]);
  }
  }
  else {
    for(int i = 1; i <= M;++i){
      A[0] = A[N - i + 1];
      A[N - i + 1] = A[N - M];
      move(A, N, M);
    }
    for(int i = 1; i <= N; ++i){
      if( i == N )
        printf("%d", A[i]);
      else printf("%d ", A[i]);
    }
  }
  return 0;
}