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

分析：所谓的循环右移，其实可以这样理解，把一串数据分成两部分，第一部分是前N-M个数据，第二部分是最后M个数据。我们要做的就是把第二部分移到第一个前面。为了不使用新数组，我们可以用旋转法来完成。比如要把123456这6个数字右移2位，那么我们先把这6个数字分成两部分，第一部分有6-2=4个，第二部分是最后2个。即1234  56。我们分别旋转这两个部分，得到4321 65。接下来再把这两部分合在一起整体旋转。432156旋转之后就变成了561234，也就是我们所要求的，有木有感到很神奇，这就就是《编程珠玑》里面提到的反掌问题。另外我感觉这道题的说明其实挺坑的，本来我考虑到了M>N这种情况，但是看到题目第一行说数据由（A0 A1……AN - 1）变换为（AN - M …… AN - 1 A0 A1……AN - M - 1），我以为他的意思是默认M>N，就把M%=N这行删掉了，结果果然报错，我再加上M%=N之后就通过啦。

#include<iostream>
using namespace std;
void rotate(int *begin, int *end)//地址传递，把从begin到end的数组翻转
{

	while (begin < end)//其实也就是把第n个和倒数第n个两两交换而已
	{
		swap(*begin, *end);//交换函数
		begin++;
		end--;
	}
}
int main()
{
	int N, M, a[101], i;
	cin >> N >> M;
	M %= N;//交换M次和交换M%N次是一样的
	for (i = 1; i <= N; i++)
		cin >> a[i];
	rotate(&a[1], &a[N-M]);//把第一部分旋转
	rotate(&a[N - M + 1], &a[N]);//把第二部分旋转
	rotate(&a[1], &a[N]);//把整体旋转
	for (i = 1; i < N; i++)
		cout << a[i] << " ";//输出前N-1个数字
	cout << a[N];//为了不多输出空格，最后一个数字单输出
}