codeup22110 拉丁方阵

codeup22110  拉丁方阵

时空限制    1000ms/128MB

题目描述

一个N×N的拉丁正方形含有整数1-N，且在任意的行或列中都不出现重复数据，一种可能的6×6拉丁正方形如下：

6 3 1 4 2 5

1 4 5 6 3 2

5 6 2 1 4 3

2 1 3 5 6 4

3 5 4 2 1 6

4 2 6 3 5 1

该拉丁方阵的产生方法是：当给出第一行数后，就决定了各数在以下各行的位置，即第一行数为各数在每行中列数的索引表（首先找到某数所在位置（下标），然后以此数在序列中的位置为起始）。

比如上例中：

第1行的第1个数为6，则该数在1—6行的列数依次为1，4，2，5，6，3（6的下标为1，在第一行序列6 3 1 4 2 5中找到1，将1排在开头重排序列则为1 4 2 5 6 3为数字6在各行的位置）

第1行的第2个数为3，则该数在1—6行的列数依次为2，5，6，3，1，4

第1行的第3个数为1，则该数在1—6行的列数依次为3，1，4，2，5，6

第1行的第4个数为4，则该数在1—6行的列数依次为4，2，5，6，3，1

第1行的第5个数为2，则该数在1—6行的列数依次为5，6，3，1，4，2

第1行的第6个数为5，则该数在1—6行的列数依次为6，3，1，4，2，5

请你写一个程序，产生按上述方法生成的拉丁方阵。

输入

第一行整数N（不大于100）；

第二行为1~N的N个数。

输出

按上述方法生成的拉丁方阵。

样例输入

6
6 3 1 4 2 5

样例输出

6 3 1 4 2 5
1 4 5 6 3 2
5 6 2 1 4 3
2 1 3 5 6 4
3 5 4 2 1 6
4 2 6 3 5 1

 

代码

#include<iostream>
using namespace std;
const int N = 105;
int n,a[N],b[N][N];

int main(){
	cin>>n;
	for (int i=0; i<n; i++) cin>>a[i];
	for (int i=0; i<n; i++){	//依次填数a[i]
		int pos;
		for (int j=0; j<n; j++)	//查找数字起始位置
			if (i+1==a[j]) { pos=j; break; }	//a[i]序号i+1在a值中位置pos
		for (int j=0; j<n; j++)	//每列依次填入
			b[j][a[(pos+j)%n]-1]=a[i];	//a下标0开始,值1开始
	}
	for (int i=0; i<n; i++){
		for (int j=0; j<n-1; j++) cout<<b[i][j]<<" ";
		cout<<b[i][n-1]<<endl;
	}
	return 0;
}