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个数进行反转，然后将其余的进行反转，最后将整个数组反转，即为答案。

举个例子

如上述样例

我们将5,6反转得到 1 2 3 4 6 5

将 1 2 3 4 反转得到 4 3 2 1 6 5

最后将整个数组反转得到答案 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?"":" ");

}
}