【PAT乙级】1008 数组元素循环右移问题 (20分)

一个数组$A$中存有$N（>0）$个整数，在不允许使用另外数组的前提下，将每个整数循环向右移$M（\ge 0）$个位置，即将A中的数据由$（A_{0}A{}_1\cdots A_N-1）$变换为$（A{}_{N-M}\cdots A{}_{N-1}A{}_0{A}_1\cdots A_{N-M-1}）$（最后$M$个数循环移至最前面的M个位置）。如果需要考虑程序移动数据的次数尽量少，要如何设计移动的方法？

输入格式:

每个输入包含一个测试用例，第1行输入$N（1\le N\le 100）$和$M（\ge 0）$；第2行输入$N$个整数，之间用空格分隔。

输出格式:

在一行中输出循环右移$M$位以后的整数序列，之间用空格分隔，序列结尾不能有多余空格。

输入样例:

6 2
1 2 3 4 5 6

输出样例:

5 6 1 2 3 4

解题思路：

矩阵转置可解决数组元素循环右移问题。我们将矩阵的后 $m$ 个元素看做 $b$，剩下的个元素看做 $a$，则原始矩阵$A=ab$，右移后的矩阵$B=ba$，我们先对A求逆得到$b^{-1}{a}^{-1}$，然后再分别对 $b^{-1}$ 和 $a^{-1}$ 求得到 $ba$。

$Note$：
(1) 当移动的个数m > 数组长度n时，m需要对n取模，即 m%=n
(2) 输出序列结尾不能有多余空格

答案代码：

#include <iostream>
#include <string>
using namespace std;

void reverse(int a[], int begin, int end) {  // 数组逆序存放
	for (int i = begin, j = end-1; i <= j; i++, j--) {
		int tmp = a[i];
		a[i] = a[j];
		a[j] = tmp; 	
	}
} 

int main() {
	int a[105] = {0}, n = 0, m = 0;
	cin >> n >> m;
	for (int i =0; i < n; i++)
		cin >> a[i];
	
    m %= n;                             // 移动的值m > 数组长度n
    
	reverse(a, 0, n);                   
	reverse(a, 0, m);
	reverse(a, m, n);
	
	for (int i =0; i < n; i++) {
		cout << a[i];
		if (i != n-1)	cout << " ";    // 序列结尾不能有多余空格
	}
	cout << endl;
	
	return 0;
}