PTA练习 7 -8 方阵循环右移（20分） （题目 + 代码 + 详细注释 ）

本题要求编写程序，将给定n×n方阵中的每个元素循环向右移m个位置，即将第0、1、⋯、n−1列变换为第n−m、n−m+1、⋯、n−1、0、1、⋯、n−m−1列。

输入格式：

输入第一行给出两个正整数m和n（1≤n≤6）。接下来一共n行，每行n个整数，表示一个n阶的方阵。

输出格式：

按照输入格式输出移动后的方阵：即输出n行，每行n个整数，每个整数后输出一个空格。

输入样例：

2 3
1 2 3
4 5 6
7 8 9

输出样例：

2 3 1 
5 6 4 
8 9 7 

//思路：我的想法是直接按一维数组读入n * n方阵的所有元素（好处是可以节省一些空间），输出时第i行和第i + 1行同一列的元素下标相差n，所以只需要算出移动后第一行的纵坐标即可。

下面是我AC的代码：

#include<cstdio>
#include<iostream>
using namespace std;
int mp[40], a[7];        //mp数组存储方阵元素，a[i]表示第i列变化后的下标
int main() {
	int m, n;
	cin >> m >> n;         //读入移动跨度m和方阵大小n
	for (int i = 1; i <= n * n; i++)
		cin >> mp[i];       //读入方阵

	for (int i = 1; i <= n; i++)           //关键
		a[(i + m - 1) % n + 1] = i;         //移动前第i列的下标为i，移动后i变为了第（i + m - 1） % n + 1列的下标

	for (int i = 0; i < n; i++) {           //i表示输出的行数
		for (int j = 1; j <= n; j++)         //j表示列数
			cout << mp[a[j] + n * i] << " ";        //n * i表示第i行第j列与第1行第j列下标之差
		cout << endl;
	}
    
	return 0;
}