【PAT_B】1008 数组元素循环右移问题（c/c++)

题目描述

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

思路：

将0到n-m-1号元素逆序，再将n-m到n-1号元素逆序，最后将整体逆序就可以得到所需序列。
reverse函数可以实现将数组或vector中的元素倒置

代码

#include<bits/stdc++.h>

using namespace std;

int main(){
    int n,m;
    cin>>n>>m;
    vector<int> a(n);
    for(int i=0;i<n;i++){
        cin>>a[i];
    }
    m %= n;
    if (m != 0) {
        reverse(begin(a), begin(a) + n);        
        reverse(begin(a) + m, begin(a) + n);
        reverse(begin(a), begin(a) + m);
    }
    for(int i=0;i<n;i++){
        if(i!=n-1){
            cout<<a[i]<<" ";
        }
        else{
            cout<<a[i];
        }        
    }
    return 0;
}