PTA 1008 数组元素循环右移问题 (20分)

一个数组A中存有N（>0）个整数，在不允许使用另外数组的前提下，将每个整数循环向右移M（≥0）个位置，即将A中的数据由（A​0​​ A​​ ⋯A​N−1）变换为（A ​N−M​​ ⋯A ​N−1​​ A​ 0 A ​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
作者
DS课程组
单位
浙江大学
代码长度限制
16 KB
时间限制
400 ms
内存限制
64 MB
编译器 (34)

帮助

#include<iostream>
using namespace std;
int main()
{
    int N,M,s,i,j;
    cin>>N>>M;
    s=M+2*N;
    int arr[s];
    for(i=0;i<s;i++)
      arr[s]=0;
    for(i=0;i<N;i++)
      cin>>arr[i];
    
    if(N>=M)
    {
      for(i=M+N;i<(M*2+N);i++)
      arr[i]=arr[i-M-M];
      
      for(i=(M*2+N);i<(M+N*2);i++)
      arr[i]=arr[i-M*2-N];
      
      for(i=M+N;i<(N*2+M-1);i++)
      cout<<arr[i]<<" ";
      cout<<arr[N*2+M-1];
    } 
    if(N<M)
    {
       j=M-N;
       for(i=j+N;i<(j*2+N);i++)
       arr[i]=arr[i-j-j];
      
       for(i=(j*2+N);i<(j+N*2);i++)
       arr[i]=arr[i-j*2-N];
      
       for(i=j+N;i<(N*2+j-1);i++)
       cout<<arr[i]<<" ";
       cout<<arr[N*2+j-1];
	}
}