蓝桥杯 九宫幻方

问题描述
　　小明最近在教邻居家的小朋友小学奥数，而最近正好讲述到了三阶幻方这个部分，三阶幻方指的是将1~9不重复的填入一个3*3的矩阵当中，使得每一行、每一列和每一条对角线的和都是相同的。

三阶幻方又被称作九宫格，在小学奥数里有一句非常有名的口诀：“二四为肩，六八为足，左三右七，戴九履一，五居其中”，通过这样的一句口诀就能够非常完美的构造出一个九宫格来。

4 9 2
3 5 7
8 1 6

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

而你呢，也被小明交付了同样的任务，但是不同的是，你需要写一个程序~ 输入格式 　　输入仅包含单组测试数据。
　　每组测试数据为一个3*3的矩阵，其中为0的部分表示被小明抹去的部分。
　　对于100%的数据，满足给出的矩阵至少能还原出一组可行的三阶幻方。 输出格式
　　如果仅能还原出一组可行的三阶幻方，则将其输出，否则输出“Too Many”（不包含引号）。
样例输入
0 7 2
0 5 0
0 3 0
样例输出
6 7 2
1 5 9
8 3 4

代码如下：

#include <stdio.h>
#include <string.h>
//map为储存输入的一维数组
int count = 0;
int result[10] = {0};//储存最终结果

int check(int *map)//判断当前九宫格是否符合要求
{
	int i, j;
	int sum1 = 0, sum2 = 0, sum3 = 0;

	sum1 = map[0] + map[1] + map[2];//先获取第一行的3个数的和

	for (i = 0; i < 3; i ++)
	{
		for (j = 0; j < 3; j ++)
		{//获取每一行每一列的3个数的和
			sum2 += map[i * 3 + j];
			sum3 += map[i + j * 3];
		}

		if (sum1 != sum2 || sum1 != sum3)//判断如果有任何一行或者任何一列和第一行的和不相等，就返回0
			return 0;
		
		sum2 = 0;
		sum3 = 0;
	}
//两个对角线的和
	sum2 = map[0] + map[4] + map[8];
	sum3 = map[2] + map[4] + map[6];

	if (sum1 != sum2 || sum1 != sum3)//判断相等
		return 0;
	
	for (i = 0; i < 9; i++)
		for (j = i + 1; j < 9; j++)
			if (map[i] == map[j])//判断是否有数字重复
				return 0;

	return 1;
}

void dfs(int *map, int pos)
{
	int i;

	while (map[pos] != 0 && pos <= 9)//如果当前数字是非0的，就让pos一直加
		pos ++;

	if (pos == 9)//说明走到了尽头，开始判断是否符合要求
	{
		if (check(map))
		{
			memcpy(result, map, 9 * sizeof(int));//符合要求的话复制map
			count ++;//用于判断结果是否有多种
		}

		return;
	}

	for (i = 1; i <= 9; i++)//暴力破解部分
	{
		if (map[pos] == 0)
		{
			map[pos] = i;
			dfs(map, pos + 1);
			map[pos] = 0;//回溯
		}
		else
			while (map[pos] != 0 && pos < 9)//跳过map[pos]!=0的地方
				pos ++;
	}
}
			
int main()
{
	int i;
	int map[10] = {0};

	for (i = 0; i < 9; i++)//输入
		scanf("%d", &map[i]);

	dfs(map, 0);

	if (count > 1)//输出
		printf("Too Many");
	else
	{
		for (i = 0; i < 9; i++)
		{
			printf("%d ", result[i]);
			if (!((i + 1) % 3))
				printf("\n");
		}
	}

	return 0;
}