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

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

分析：

题目要要求我们原地完成数组的移动，可以分为两种情况：

1. step 是 N的倍数则需要移动。
2. step 不是 N的倍数则不需要移动。

以样例来说可以模拟其移动过程：

1 2 3 4 5 6 -> 6 5 4 3 2 1 -> 5 6 4 3 2 1 -> 5 6 1 2 3 4

按照模拟我们不难看出数组移动的顺序是：

1. 逆转整个数组
2. 再逆转从 0 到 step - 1的位置
3. 最后再逆转从step到数组尾部

得到这个结论就可以开始写代码了

代码：

#include <iostream>
#include <algorithm>

using namespace std;

int main() {
    int N, step;
    cin >> N >> step;
    int arr[N];
    for (int i = 0; i < N; i++)
        cin >> arr[i];
    
    if (step % N != 0) {	// 如果 step 不是 N 的倍数
        step %= N;
        reverse(arr, arr + N);			// 逆转整个数组
        reverse(arr, arr + step);		// 逆转 0 到 step - 1
        reverse(arr+ step, arr + N);	// 逆转 step 到数组末尾
        for (int i = 0; i < N - 1; i++)
            cout << arr[i] << " ";
        cout << arr[N - 1];
    } else {	
        // 直接输出
        for (int i = 0; i < N - 1; i++)
            cout << arr[i] << " ";
        cout << arr[N - 1];
    }
    printf("\n");
    return 0;
}