hihoCoder 1268 九宫

#1268 : 九宫

时间限制: 10000ms

单点时限: 1000ms

内存限制: 256MB

描述

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

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

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

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

输入

输入仅包含单组测试数据。

每组测试数据为一个3*3的矩阵，其中为0的部分表示被小Hi抹去的部分。

对于100%的数据，满足给出的矩阵至少能还原出一组可行的三阶幻方。

输出

如果仅能还原出一组可行的三阶幻方，则将其输出，否则输出“Too Many”（不包含引号）。

样例输入

0 7 2
0 5 0
0 3 0

样例输出

6 7 2
1 5 9
8 3 4

这种题看一眼都知道是搜索。

方法DFS。

算比较简单的DFS了。

开一个数组vis，记录当前1-9数字中哪些出现了，那些没出现。

DFS搜索过程中，枚举那些还没出现的数字

当如果判断条件符合，则返回，输出结果

如果条件不符，回溯，将vis[i]再置为0，把graph[i]也置为未填。

判断函数就是枚举行列，对角线上的数和是否相等。

#include <stdio.h>
#include <string.h>
const int MAXN=10;
int graph[MAXN],vis[MAXN],ans[MAXN];
int flag=0;
bool isok()
{
        int sum=graph[1]+graph[2]+graph[3];
        for(int i=4;i<=9;i+=3)
                if((graph[i]+graph[i+1]+graph[i+2])!=sum)
                        return false;
        for(int i=1;i<=3;i++)
                if((graph[i]+graph[i+3]+graph[i+6])!=sum)
                return false;
        if((graph[1]+graph[5]+graph[9])!=sum||(graph[3]+graph[5]+graph[7])!=sum)
                return false;
        return true;
}

void dfs(int pos)
{
        if(pos==10&&isok())
        {
                flag++;
                if(flag==1)
                        memcpy(ans,graph,sizeof(graph));
                return;
        }
        if(graph[pos])
                dfs(pos+1);
        else
                for(int i=1;i<=9;i++)
                {
                        if(vis[i])
                                continue;
                        vis[i]=1;
                        graph[pos]=i;
                        dfs(pos+1);
                        vis[i]=0;
                        graph[pos]=0;
                }
}
int main()
{
        memset(vis,0,sizeof(vis));
        for(int i=1;i<=9;i++)
        {
                scanf("%d",&graph[i]);
                vis[graph[i]]=1;
        }
        dfs(1);
        if(flag==1)
                for(int i=1;i<=9;i++)
                      printf("%d%c", ans[i], (i%3==0)? '\n' : ' ');
        else if(flag>1)
                printf("Too Many\n");
        return 0;
}