1334: [蓝桥杯2017初赛]九宫幻方【简单】 有点意思

http://oj.ecustacm.cn/problem.php?id=1334
思路:
其实就是一个全排列问题，不过有了一点的限制。
我们先用 b 数组 来保存我们的输入的数组。
再用a数组来保存 1 ~ 9的全排列。
每列出一种排列就要判断和 b 数组中非零的数是不是全部的一样。
接着判断 行列对角线是不是相等。
其实有一个规律是，三层幻方的 相等的值是 15， 所以只要判断各个边的和是不是15就行了。

#include<cstdio>
#include<iostream>
#include<algorithm> 
using namespace std;
int a[9]={1,2,3,4,5,6,7,8,9};
int b[9];
int sum1,sum2,sum3;
int sum4,sum5,sum6;
int sum7,sum8;
int main(void)
{
	int n=0;
	for(int i=0;i<9;i++)
	{
		scanf("%d",&b[i]);
	}
	do
	{
		bool flag=true;
		for(int i=0;i<9;i++)
		{
			if(b[i]!=0)
			{
				if(b[i]!=a[i])
				{
					flag=false;
					break;
				}
			}
		}
		if(flag)
		{
			sum1=a[0]+a[1]+a[2];
			sum2=a[3]+a[4]+a[5];
			sum3=a[6]+a[7]+a[8];
			sum4=a[0]+a[3]+a[6];
			sum5=a[1]+a[4]+a[7];
			sum6=a[2]+a[5]+a[8];
			sum7=a[0]+a[4]+a[8];
			sum8=a[2]+a[4]+a[6];
			if(sum1==15&&sum2==15&&sum3==15&sum4==15&&sum6==15&&sum7==15&&sum8==15)
			{
				for(int i=0;i<9;i++)
				{
					b[i]=a[i];//保存幻方的值
				}
				n++;
			}
		}
	}while(next_permutation(a,a+9));
	if(n==1)
	{
		for(int i=1;i<=9;i++)
		{
			printf("%d ",b[i-1]);
			if(i%3==0)
			{
				printf("\n");
			}
		}
	}
	else
	{
		printf("Too Many\n");
	}
	return 0;
} 

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
int a[9]={1,2,3,4,5,6,7,8,9};
int b[9];
int ans;
int main(void)
{
	for(int i=0;i<9;i++) cin>>b[i];
	do
	{
		bool flag=true;
		
		if(ans>1) break;	//答案有两次了 
		int sum=0;
		for(int i=0;i<3;i++) sum+=a[i];//第一行 
		if(sum!=15) continue;
		sum=0;
		for(int i=3;i<6;i++) sum+=a[i];//第二行 
		if(sum!=15) continue;
		sum=0;
		for(int i=6;i<9;i++) sum+=a[i];//第三行 
		if(sum!=15) continue;
		sum=0;
		sum=a[0]+a[3]+a[6];//第一列 
		if(sum!=15) continue;
		sum=0;
		sum=a[1]+a[4]+a[7];//第二列 
		if(sum!=15) continue;
		sum=0;
		sum=a[2]+a[5]+a[8];//第三列 
		if(sum!=15) continue;
		sum=0;
		sum=a[0]+a[4]+a[8];//做对角 
		if(sum!=15) continue;
		sum=0;
		sum=a[2]+a[4]+a[6];//右对角 
		if(sum!=15) continue;
		for(int i=0;i<9;i++)
		{
			if(b[i]!=0&&b[i]!=a[i])//位置不匹配 
				flag=false;
		}
		if(ans==0&&flag)//第一次 
		{
			for(int i=0;i<9;i++)
			{
				b[i]=a[i];
			} 
			ans++;
		}
	}while(next_permutation(a,a+9));
	if(ans==1)
		for(int i=0;i<9;i++)
		{
			cout<<b[i]<<" ";
			if((i+1)%3==0) cout<<endl; 
		}
	else
	cout<<"Too Many";
	return 0;
}