蓝桥杯——九宫幻方

题目

小明最近在教邻居家的小朋友小学奥数，而最近正好讲述到了三阶幻方这个部分，
三阶幻方指的是将1~9不重复的填入一个3*3的矩阵当中，使得每一行、每一列和每一条对角线的和都是相同的。
三阶幻方又被称作九宫格，在小学奥数里有一句非常有名的口诀：
“二四为肩，六八为足，左三右七，戴九履一，五居其中”，
通过这样的一句口诀就能够非常完美的构造出一个九宫格来。

4 9 2
3 5 7
8 1 6
有意思的是，所有的三阶幻方，都可以通过这样一个九宫格进行若干镜像和旋转操作之后得到。
现在小明准备将一个三阶幻方（不一定是上图中的那个）中的一些数抹掉，交给邻居家的小朋友来进行还原，
并且希望她能够判断出究竟是不是只有一个解。

而你呢，也被小明交付了同样的任务，但是不同的是，你需要写一个程序~

输入
0 7 2
0 5 0
0 3 0

输出
6 7 2
1 5 9
8 3 4

思路

该题用dfs解决，vis数组标记该数是不是有了，p 用来获取坐标，cnt为一则是唯一的，check函数来判断是不是九宫幻方，在dfs中当 now>n时表示九个数已经填完了，然后用check判断，如果now <n 则填数字。

代码

#include<iostream>
using namespace std;
int vis[10],a[5][5],ans[5][5];//cnt数组标记这个数出现没，ans用来输出 
int cnt,n;//cnt数组用来判断三阶幻方是不是唯一的 
pair<int,int> p[10]; //获取该位置的坐标 
bool check() //判断是不是三阶幻方 
{
	int sum = a[1][1]+a[2][2]+a[3][3];
	if(a[1][3]+a[2][2]+a[3][1] != sum) return false;
	for(int i = 1;i < 4;i++)
	{
		int tmp1 = 0, tmp2 = 0; //判断行和列的符合题意的 
		for(int j = 1;j < 4;j++)
		{
			tmp1 += a[i][j],tmp2 += a[j][i];
		}
		if(tmp1 != sum||tmp2 != sum) return false;
	}
	return true; //
}
void dfs(int now)
{
	if(now > n) //已经全部填充了 
	{
		if(check())
		{
			cnt++;
			for(int i = 1;i < 4;i++)
			{
				for(int j = 1;j < 4;j++)
				{
					ans[i][j] = a[i][j];
				}
			}
			return ;
		}
	}
	int x = p[now].first ,y = p[now].second;  //该点的坐标 
	for(int j =1;j < 10;j++)
	{
		if(vis[j]) continue; //该数已经有了 
		a[x][y] = j;
		vis[j] = 1;
		dfs(now+1);
		a[x][y] = 0; //表示回溯 
		vis[j] = 0;
		
	}
}
int main()
{
    for(int i = 1;i < 4;i++)
    {
    	for(int j = 1;j < 4;j++)
    	{
    		cin>>a[i][j];
    		if(!a[i][j])p[++n] = make_pair(i,j);//放p里面 
    		vis[a[i][j]] == 1;  //标记该点已经有了 
		}
	}
	dfs(1);
	if(cnt == 1) //唯一的输出 
	{
		for(int i = 1;i < 4;i++)
		{
			for(int j = 1;j < 4;j++)
			{
				cout<<ans[i][j]<<" \n"[j == 3];
			}
		}
	}
	else cout<<"Too Many\n";
	return 0;
}

• 总结：关键是用dfs解决问题，创建vis数组判断该数字是不是已经有了，和p来获取坐标