Hihocoder1268九宫格问题【dfs基础】

九宫格填充问题【dfs基础】

九宫格是指3*3的幻方，其中填入1-9共九个数字，使得每一行、每一列、每一条对角线的和都是相同的。

构造九宫格有句口诀：“二四为肩，六八为足，左三右七，戴九履一，五居其中”，这样就可以构造出一个完美的九宫格。
有趣的是，所有的九宫格都可以由上述的基本九宫格通过旋转和镜像来获得。现在有一个九宫格，其中未填充的数用0来表示。如果该九宫格有多组解，就输出“Too Many”，否则，输出还原的那组解。

Sample Input
0 7 2
0 5 0
0 3 0

Sample Output
6 7 2
1 5 9
8 3 4

【思路】
已知一个九宫格的部分数据，来还原整个九宫格，由于数据规模较小，可以考虑dfs来解，用全局answers来记录解的数目即可。注意dfs时，如果找到了一组可行解，需要用ans数组来保存下当前的九宫格，否则当前dfs完成后，mp中的原本未知数据还是会被抹去。
以下是AC代码：

#include <bits/stdc++.h>
using namespace std;
int mp[10];
int ans[10];
bool vis[10];
int answers;

bool check()
{
    int s1 = mp[1] + mp[2] + mp[3];
    int s2 = mp[4] + mp[5] + mp[6];
    int s3 = mp[7] + mp[8] + mp[9];
    int s4 = mp[1] + mp[4] + mp[7];
    int s5 = mp[2] + mp[5] + mp[8];
    int s6 = mp[3] + mp[6] + mp[9];
    int s7 = mp[1] + mp[5] + mp[9];
    int s8 = mp[3] + mp[5] + mp[7];

    if(s1 == s2 && s2 == s3 && s3 == s4 && s4 == s5 &&
       s5 == s6 && s6 == s7 && s7 == s8)
        return true;
    else
        return false;
}

void dfs(int pos)
{
    if(pos == 10 && check()){
        answers++;
        if(answers == 1){
            for(int i = 1; i <= 9; i++)
                ans[i] = mp[i];
        }
        return ;
    }
    else if(pos < 10){
        if(mp[pos])
            dfs(pos+1);
        else {
            for(int i = 1; i <= 9; i++){
                if(vis[i]) continue;
                vis[i] = true;
                mp[pos] = i;
                dfs(pos+1);
                vis[i] = false;
                mp[pos] = 0;
            }
        }
    }
}

int main()
{
    for(int i = 1; i <= 9; i++){
        cin >> mp[i];
        vis[mp[i]] = true;
    }
    dfs(1);
    if(answers == 1){
        for(int i = 1; i <= 9; i++){
            if(i % 3 == 0) cout << ans[i] << endl;
            else cout << ans[i] << " ";
        }
    }
    else {
        cout << "Too Many" << endl;
    }
    return 0;
}