PAT乙级-1008 数组元素循环右移问题 (20分)

点击链接PAT乙级-AC全解汇总

题目：
一个数组A中存有N（>0）个整数，在不允许使用另外数组的前提下，将每个整数循环向右移M（≥0）个位置，即将A中的数据由（A₀ A₁⋯A_N-1​​ ）变换为（A_N-M ⋯A_N-1A₀ A₁ ⋯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<iostream>
#include<string>
#include<algorithm>
#include<cstring>
using namespace std;

int main()
{
    int M,N;
    cin>>N>>M;
    M=M%N;
    int a[N];
    for(int i=0;i<N;i++) cin>>a[i];

    if(M==0)
    {
        cout<<a[0];
        for(int i=1;i<N;i++)cout<<" "<<a[i];
        return 0;
    }
    //step1
    int id_start=0;
    int id_end=N-M-1;
    for(int i=id_start;i<(id_start+id_end+1)/2;i++)
    {
        int t=a[i];
        a[i]=a[id_start+id_end-i];
        a[id_start+id_end-i]=t;
    }
    //step2
    id_start=N-M;
    id_end=N-1;
    for(int i=id_start;i<(id_start+id_end+1)/2;i++)
    {
        int t=a[i];
        a[i]=a[id_start+id_end-i];
        a[id_start+id_end-i]=t;
    }
    //step3
    id_start=0;
    id_end=N-1;
    for(int i=id_start;i<(id_start+id_end+1)/2;i++)
    {
        int t=a[i];
        a[i]=a[id_start+id_end-i];
        a[id_start+id_end-i]=t;
    }

    cout<<a[0];
    for(int i=1;i<N;i++)cout<<" "<<a[i];
    return 0;
}

循环右移的思路就是左边逆序、右边逆序、整体逆序

化简函数版本：

#include<bits/stdc++.h>
using namespace std;

int a[100];
void my_reverse(int id_start,int id_end){
    for(int i=id_start;i<(id_start+id_end+1)/2;i++){
        int t=a[i];
        a[i]=a[id_start+id_end-i];
        a[id_start+id_end-i]=t;
    }
}

int main()
{
    int M,N;
    cin>>N>>M;
    M=M%N;
    for(int i=0;i<N;i++) cin>>a[i];

    if(M==0)
    {
        cout<<a[0];
        for(int i=1;i<N;i++)cout<<" "<<a[i];
        return 0;
    }

    my_reverse(0,N-M-1);
    my_reverse(N-M,N-1);
    my_reverse(0,N-1);

    cout<<a[0];
    for(int i=1;i<N;i++)cout<<" "<<a[i];
    return 0;
}