[PAT乙级]1008. 数组元素循环右移问题 (20)

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

原题链接
一个数组A中存有N（N>0）个整数，在不允许使用另外数组的前提下，将每个整数循环向右移M（M>=0）个位置，即将A中的数据由（A0 A1……AN-1）变换为（AN-M …… AN-1 A0 A1……AN-M-1）（最后M个数循环移至最前面的M个位置）。如果需要考虑程序移动数据的次数尽量少，要如何设计移动的方法？

输入格式：每个输入包含一个测试用例，第1行输入N ( 1<=N<=100)、M（M>=0）；第2行输入N个整数，之间用空格分隔。

输出格式：在一行中输出循环右移M位以后的整数序列，之间用空格分隔，序列结尾不能有多余空格。

输入样例：
6 2
1 2 3 4 5 6
输出样例：
5 6 1 2 3 4

思路：

• 这道题我刚开始没想到，后来查看别人代码，人家用到了reverse()函数，我就查了查这个函数的用法，点击查看reverse()函数用法
• reverse()函数头文件 #include 《algorithm>
• 有了这个函数，题目变得很简单，给出数组长度N，右移位数M，右移后的数组其实就是把原数组后面M个元素提到数组前面，用三次reverse()函数就能完成转换，看下面的代码实现

代码：

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

int main()
{
    int N,M;
    cin >> N >> M;
    if(M > N)
        M -= N;
    vector<int> arr(N);
    for(int i=0; i<N; i++){
        int temp;
        cin >> temp;
        arr[i] = temp;
    }
    reverse(arr.begin(),arr.end());
    reverse(arr.begin(),arr.begin()+M);//从数组开头查M个元素转换顺序
    reverse(arr.begin()+M,arr.end());//从数组第M+1个元素到数组末尾转换顺序，arr.end()可以写成arr.begin()+N
    for(int j=0; j<N; j++){
        if(j != 0)
            cout << " ";
        cout << arr[j];
    }
    return 0;
}