hiho 第135周 九宫

题目链接：http://hihocoder.com/contest/hiho135/problem/1

由于是九宫格，全排列也就是9! (362880)种方式，我就直接暴力枚举排列好了。

#include <bits/stdc++.h>

using namespace std;

int a[3][3];
int b[3][3];
bool vis[10];
int num[10];
int cnt;

bool check() {
    int num = b[0][0] + b[0][1] + b[0][2];
    int num2 = 0;
    for(int i=0;i<3;i++)
        num2 +=b[1][i];
    if(num2!=num) return false;

    num2 = 0;
    for(int i=0;i<3;i++)
        num2+=b[2][i];
    if(num2!=num) return false;

    for(int i=0;i<3;i++)
    {
        num2 = 0;
        for(int j=0;j<3;j++)
        {
            num2 +=b[j][i];
        }
        if(num2!=num)
            return false;
    }

    num2 = 0;
    for(int i=0;i<3;i++)
        num2 +=b[i][i];
    if(num2!=num)
        return false;

    num2 = 0;
    for(int i=0;i<3;i++)
        num2 +=b[i][2-i];
    if(num2!=num)
        return false;

    return true;
}

int main()
{
    cnt = 0;
    for(int i=0;i<3;i++)
        for(int j=0;j<3;j++) {
            scanf("%d",&a[i][j]);
            if(a[i][j])
                vis[a[i][j]] = true;
        }


    for(int i=1;i<=9;i++)
        if(!vis[i]) {
            num[cnt++] = i;
        }

    vector <int> ans[10000];
    int ansnum = 0;

    do {
        int k = 0;
        for(int i=0;i<3;i++) {
            for(int j=0;j<3;j++) {
                if(a[i][j]==0)
                    b[i][j]=num[k++];
                else b[i][j] = a[i][j];
            }
        }

        if(check())
        {
            for(int i=0;i<3;i++)
                for(int j=0;j<3;j++)
                    ans[ansnum].push_back(b[i][j]);
            ansnum++;
        }

    }while(next_permutation(num,num+cnt));

    if(ansnum>1)
        puts("Too Many");
    else if(ansnum==1)
    {
        int k = 0;
        for(int i=0;i<3;i++)
        {
            for(int j=0;j<3;j++)
            {
                printf("%d ",ans[0][k++]);
            }
            puts("");
        }
    }
    return 0;
}