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

代码

#include <iostream>
using namespace std;
int main() {
    int n, m;
    cin >> n >> m;
    int a[n],b[n],count = 0;
    for (int i = 0; i < n; i++)
        cin >> a[i];
    if (m != n && m != 0) {
        if (m > n) m = m - n;
        for(int i = n-m; i < n; ++i) {
            b[count++] = a[i];
        }
        for(int i = 0; i < n-m; ++i) {
            b[count++] = a[i];
        }
    } else {
        for (int i = 0; i < n; i++)
            b[i] = a[i] ;
    }
    for (int i = 0; i < n - 1; i++)
        cout << b[i] << ' ';
    cout << b[n - 1];
    return 0;
}