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

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

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

分析

  一开始想偷懒，直接从数组后面开始输出。发现答案有错。一看题目发现并没有说M一定小于N。不要想当然！
  于是我踩下了第二个坑：有两组测试数据显示格式错误。后来发现原因在于我用的循环中的flag确定格式。而这个结构只在第一个输出循环中出现。一旦出现极端数据，例如M=0，第一个输出循环就会被跳过，输出中的第一个数前面会多一个空格。解决方法很简单，在第二个输出循环里面加上flag结构即可。
  给我的教训就是：如果时间允许的话，严格按照题目要求来写，就不会踩坑。

代码

#include <iostream>
#include <cmath>
using namespace std;
int main()
{
	int n, m, flag = 0;
	int a[105];
	cin >> n >> m;
	for(int i = 0; i < n; i++)
		cin >> a[i];
	if(m >= n)
	{
		m = m % n;
	}
	for(int i = n - m; i < n; i++)
	{
		if(flag == 0)
		{
			cout << a[i];
			flag++;
		}
		else
			cout << ' ' << a[i];
	}
	for(int i = 0; i < n - m; i++)
	{
		if(flag == 0)
		{
			cout << a[i];
			flag++;
		}
		else
		cout << ' ' << a[i];
	}
	return 0;
}