蓝桥杯 九宫幻方(穷竭搜索之模拟)

题目

输入三阶幻方，没有填数的地方，输入0。

如果仅能还原出一组可行的三阶幻方，则将其输出，

否则输出“Too Many”（不包含引号）。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=1e5+10;
int num[3][3],res[3][3];
int mp[3][3],ans;
int l[10],cnt,now;
bool vis[10];
bool check(int a[3][3])
{
	int p=0,sum[10]={0};
	for(int i=0;i<3;++i)
	for(int j=0;j<3;++j)
	sum[i]+=a[i][j];
	for(int i=0;i<3;++i)
	for(int j=0;j<3;++j)
	sum[3+j]+=a[i][j];
	for(int i=0;i<3;++i)
	{
		sum[6]+=a[i][i];
		sum[7]+=a[i][2-i];
	}
	for(int i=1;i<8;++i)
	if(sum[i]!=sum[0])return 0;
	return 1;
}
void output(int res[3][3])
{
	for(int i=0;i<3;++i)
	for(int j=0;j<3;++j)
	printf("%d%c",res[i][j],j==2?'\n':' ');
}
int main()
{
	for(int i=0;i<3;++i)
	{
		for(int j=0;j<3;++j)
		{
			scanf("%d",&mp[i][j]);
			if(mp[i][j])vis[mp[i][j]]=1;
		}
	}
	for(int i=1;i<=9;++i)
	if(!vis[i])l[cnt++]=i;
	do
	{
		now=0;
		for(int i=0;i<3;++i)
		for(int j=0;j<3;++j)
		num[i][j]=mp[i][j];
		for(int i=0;i<3;++i)
		{
			for(int j=0;j<3;++j)
			{
				if(!num[i][j])
				num[i][j]=l[now++];
			} 
		}
		if(check(num))
		{
		 for(int i=0;i<3;++i)
		 for(int j=0;j<3;++j)
		 res[i][j]=num[i][j];
		 ans++;
	    }
		if(ans>=2)break;
	}while(next_permutation(l,l+cnt));
	if(ans==1)output(res);
	else puts("Too Many");
	return 0;
}