poj 2918 Tudoku

数独。

题目链接：点击打开链接

#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
using namespace std;

char mp[10][10];
bool mk[10][10];
vector<int> h[10];
vector<int> s[10];

void solve()
{
    for(int i=1;i<=9;i++)
    {
        for(int j=1;j<=9;j++)
        {
            int flag=1;
            for(int k=1;k<=9;k++)
            {
                if(mp[j][k]==i+'0')
                    flag=0;
            }
            if(flag)
                h[j].push_back(i);
        }
    }
    for(int i=1;i<=9;i++)
    {
        for(int j=1;j<=9;j++)
        {
            int flag=1;
            for(int k=1;k<=9;k++)
            {
                if(mp[k][j]==i+'0')
                    flag=0;
            }
            if(flag)
                s[j].push_back(i);
        }
    }
}

int dfs(int num)
{
    int tmp;
    if(num==0) return 1;
    for(int i=1;i<=9;i++)
    {
        for(int j=1;j<=9;j++)
        {
            if(mk[i][j])
            {
                for(int k=0;k<h[i].size();k++)
                {
                    for(int p=0;p<s[j].size();p++)
                    {
                        if(h[i][k]==s[j][p] && h[i][k]!=0)
                        {
                            tmp=h[i][k];
                            mp[i][j]=tmp+'0';
                            h[i][k]=0;
                            s[j][p]=0;
                            mk[i][j]=false;

                            if(dfs(num-1))
                                return 1;

                            h[i][k]=tmp;
                            s[j][p]=tmp;
                            mk[i][j]=true;
                            mp[i][j]='0';
                        }
                    }
                }
                return 0;
            }
        }
    }
    return 0;
}

int main()
{
    int n,ca,tmp;
    scanf("%d",&n);
    for(int ca=1;ca<=n;ca++)
    {
        memset(mk,0,sizeof(mk));
        memset(mp,0,sizeof(mp));

        int cnt=0;
        for(int i=1;i<=9;i++)
        {
            scanf("%s",mp[i]+1);
            for(int j=1;j<=9;j++)
            {
                tmp=mp[i][j]-'0';
                if(tmp==0)
                {
                    mk[i][j]=true;
                    cnt++;
                }
            }
        }


        if(cnt==0)
        {
            if(ca!=1) putchar('\n');
            printf("Scenario #%d:\n",ca);
            for(int i=1;i<=9;i++)
                printf("%s\n",mp[i]+1);
        }
        else
        {
            if(ca!=1) putchar('\n');
            printf("Scenario #%d:\n",ca);
            solve();
            dfs(cnt);
            for(int i=1;i<=9;i++)
                printf("%s\n",mp[i]+1);
        }
    }
    return 0;
}