问题 BC: 拉丁方阵

题目描述

拉丁方阵 （Latin square）是一种 n × n 的 方阵 ，在这种 n × n 的方阵里，恰有 n 种不同的元素 ，每一种不同的元素在同一行或同一列里只出现一次。
输入一个整数N，输出所有的拉丁方阵（排头数字小的拉丁方阵优先输出）。

输入格式

一个整数N，1 <= N <=10

输出格式

所有N阶拉丁方阵

输入样例1

2

输出样例1

1 2
2 1

2 1
1 2

输入样例2

3

输出样例2

1 2 3 
2 3 1 
3 1 2 

2 3 1 
3 1 2 
1 2 3 

3 1 2 
1 2 3 
2 3 1 

代码展示 

几个需要注意的地方:

①虽然题目说的是输出“所有N阶拉丁方阵”，但博主检查未通过用例时发现，其实输入为n的时候只需要输出n个N阶拉丁方阵，分别是排头数字为1到n的情况（并且是特殊情况）。

②观察用例的输出发现，第一个矩阵的第一行一定是1到n顺序排列，接下来的几行，可以看成是上一行左移一位得到（首尾相接，对n取模实现），于是利用这个特性可以双重循环构造出第一个拉丁方阵（下文称此方阵为基础矩阵）。

③再观察，第二个矩阵可以看成是基础矩阵从第二行开始输出，第一行被挪到最后得到的，后面的矩阵以此类推（也就是代码中k循环的作用，行索引（i+k）对n取模可看成矩阵的行是首尾相接的）。

④按照上述思路写成代码，完成^^

#include <bits/stdc++.h>
using namespace std;

int main()
{
    //freopen("/config/workspace/test/test","r",stdin);
    int n;
    cin>>n;
    int a[n][n];
    for(int i=0;i<n;i++){//第一行是n的顺序全排列
        a[0][i]=i+1;
    }
    for(int i=1;i<n;i++){//基础矩阵构造
        for(int j=0;j<n;j++){
            a[i][j]=a[i-1][(j+1)%n];//每一行依赖上一行生成
        }
    }
    for(int k=0;k<n;k++){//k的作用是规定从哪一行开始输出（基础矩阵）
        for(int i=k;i<n+k;i++){//从第k行开始输出，共输出n行
            for(int j=0;j<n;j++){
                cout<<a[i%n][j]<<" ";//行对n取模，以保证不会超范围（实际上相当于首尾相接）
            }
            cout<<endl;
        }
        cout<<endl;
    }
    return 0;
}