hihoCoder - 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）+枚举判断。

#include <cstdio>
#include <algorithm>

#define MAX 5

using namespace std;

int MGraph[MAX][MAX];
int Matrix[MAX][MAX];
int ans[MAX][MAX];
int visit[10];
int num[10];

int cnt = 0;

void initMatrix() {
    for( int i = 1; i <= 3; i++ )
        for( int j = 1; j <= 3; j++ )
            if( MGraph[i][j] != 0 ) Matrix[i][j] = MGraph[i][j];
            else Matrix[i][j] = 0;
}

bool check() {
    initMatrix();

    int k = 0;
    for( int i = 1; i <= 3; i++ ) {
        for( int j = 1; j <= 3; j++ ) {
            if( Matrix[i][j] == 0 ) {
                Matrix[i][j] = num[k];
                k++;
            }
        }
    }

    if( Matrix[1][1] + Matrix[1][2] + Matrix[1][3] != 15 ) return false;
    if( Matrix[2][1] + Matrix[2][2] + Matrix[2][3] != 15 ) return false;
    if( Matrix[3][1] + Matrix[3][2] + Matrix[3][3] != 15 ) return false;
    if( Matrix[1][1] + Matrix[2][1] + Matrix[3][1] != 15 ) return false;
    if( Matrix[1][2] + Matrix[2][2] + Matrix[3][2] != 15 ) return false;
    if( Matrix[1][3] + Matrix[2][3] + Matrix[3][3] != 15 ) return false;
    if( Matrix[1][1] + Matrix[2][2] + Matrix[3][3] != 15 ) return false;
    if( Matrix[1][3] + Matrix[2][2] + Matrix[3][1] != 15 ) return false;

    return true;

}

void copy() {
    for( int i = 1; i <= 3; i++ )
        for( int j = 1; j <= 3; j++ )
            ans[i][j] = Matrix[i][j];
}

void print() {
    for( int i = 1; i <= 3; i++ ) {
        for( int j = 1; j <= 3; j++ ) {
            printf( "%d ", ans[i][j] );
        }
        printf( "\n" );
    }
}

int main() {
    for( int i = 1; i <= 3; i++ ) {
        for( int j = 1; j <= 3; j++ ) {
            scanf( "%d", &MGraph[i][j] );
            if( MGraph[i][j] != 0 ) visit[MGraph[i][j]] = 1;
        }
    }

    int j = 0;
    for( int i = 1; i <= 9; i++ ) {
        if( visit[i] == 0 ) num[j++] = i;
    }

    do {
        bool flag = check();
        if( flag ) {
            copy();
            cnt++;
        }
        if( cnt > 1 ) break;
    } while( next_permutation( num, num + j ) );

    if( cnt > 1 ) {
        printf( "Too Many\n" );
    }
    else {
        print();
    }

    return 0;
}