填幻方题解

[GESP样题 四级] 填幻方

题目描述

在一个 $N\times N$ 的正方形网格中，每个格子分别填上从 1 到 $N\times N$ 的正整数，使得正方形中任一行、任一列及对角线的几个数之和都相等，则这种正方形图案就称为“幻方”（输出样例中展示了一个 $3\times 3$ 的幻方）。我国古代称为“河图”、“洛书”，又叫“纵横图”。

幻方看似神奇，但当 $N$ 为奇数时有很方便的填法：

1. 一开始正方形中没有填任何数字。首先，在第一行的正中央填上 $1$。
2. 从上次填数字的位置向上移动一格，如果已经在第一行，则移到同一列的最后一行；再向右移动一格，如果已经在最右一列，则移动至同一行的第一列。如果移动后的位置没有填数字，则把上次填写的数字的下一个数字填到这个位置。
3. 如果第 2 步填写失败，则从上次填数字的位置向下移动一格，如果已经在最下一行，则移到同一列的第一行。这个位置一定是空的（这可太神奇了！）。把上次填写的数字的下一个数字填到这个位置。
4. 重复 2、3 步骤，直到所有格子都被填满，幻方就完成了！

快来编写一个程序，按上述规则，制作一个 $N\times N$ 的幻方吧。

输入格式

输入为一个正奇数 $N$，保证 $3\le N\le 21$。

输出格式

输出 $N$ 行，每行 $N$ 个空格分隔的正整数，内容为 $N\times N$ 的幻方。

样例 #1

样例输入 #1

3

样例输出 #1

8 1 6
3 5 7
4 9 2

思路

简单来说，除去第一个数字，所有数字都是先尝试是否能填在上一个数的右上方（行-1，列+1）的位置，并处理超限情况，如果右上方已经有数了，那么就填在上一个数的正下方。

实现

#include<bits/stdc++.h>
using namespace std;
int n,px,py,x,y,a[25][25];
//n:幻方的大小 px:上一个数的横坐标 py:上一个数的纵坐标
//x:当前数的横坐标 y:当前数的纵坐标 a:存储幻方每个位置上的数
int main()
{
	cin>>n;//读入幻方的大小
	a[1][(n+1)/2]=1;//将1填在第一行的正中央
	px=1;//更新px
	py=(n+1)/2;//更新py
	for(int i=2;i<=n*n;i++)//枚举从2到n*n的所有数，也就是幻方中要填的所有数
	{
		x=px-1;
		y=py+1;//先将这个数的位置设定在上个数的右上方，尝试一下
		if(x==0)x=n;
		else if(x==n+1)x=1;//处理x超限的情况
		if(y==0)y=n;
		else if(y==n+1)y=1;//处理y超限的情况
		if(a[x][y]!=0)x=px+1,y=py;//如果右上方有数了，那么填在正下方
		a[x][y]=i;//填数
		px=x;
		py=y;//更新px，py
	}
	for(int i=1;i<=n;i++)//输出
	{
		for(int j=1;j<=n;j++)cout<<a[i][j]<<" ";
		cout<<endl;
	}
	return 0;
}