PAT 乙级 经典问题 1008. 数组元素循环右移问题

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

一个数组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

---

解题思路

最简单的思路就是用另一个数组来对当前数组进行操作。这个题有一个很有技巧的解法，将数组末尾的M个数进行反转，然后将其余的进行反转，最后将整个数组反转，即为答案。

举个例子

如上述样例

1. 我们将5,6反转得到 1 2 3 4 6 5
2. 将 1 2 3 4 反转得到 4 3 2 1 6 5
3. 最后将整个数组反转得到答案 5 6 1 2 3 4

这个方法可以在原数组上进行，不需要任何额外的空间，并且复杂度为$O(n)$

代码

#include <cstdio>
void reverse(int *a, int n) { //n为数组长度，a为数组的首地址
    //传入进去的a数组将发生变
    int t;
    for (int i = 0; i < n / 2; i++) {
        t = a[i];
        a[i] = a[n - i - 1];
        a[n - i - 1] = t;
    }
}
int main() {
    int n,m,a[110];
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;i++){
        scanf("%d",&a[i]);
    }
    m%=n;  //别忘了考虑m>n的情况
    reverse(a+n-m,m);
    reverse(a,n-m);
    reverse(a,n);
    for(int i=0;i<n;i++){
        printf("%d%s",a[i],i==n-1?"":" ");

    }
}