题目
输入三阶幻方,没有填数的地方,输入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;
}