HDU - 5547 - Sudoku (dfs)

思路:

纯暴力会超时，每次填的时候需要判断当前行列是否满足情况，最后判断2x2的是否满足即可

ps：一定不要忘了回溯

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <set>
#include <queue>
using namespace std;
const int maxn = 10;
int n;
int sx[20];
int sy[20];
int ma[maxn][maxn];
int cnt;
int a[4] = {1,2,3,4};
set<int>se;
void test()
{
    for(int i  = 0; i <4; i++)
    {
        for(int j = 0 ; j <4; j++)
        {
            cout<<ma[i][j];
        }
        cout<<endl;
    }
}
bool check()
{
    se.clear();
    se.insert(ma[0][0]);
    se.insert(ma[1][0]);
    se.insert(ma[0][1]);
    se.insert(ma[1][1]);
    if(se.size()!=4) return 0;
    se.clear();
    se.insert(ma[0][2]);
    se.insert(ma[1][2]);
    se.insert(ma[0][3]);
    se.insert(ma[1][3]);
    if(se.size()!=4) return 0;
    se.clear();
    se.insert(ma[2][0]);
    se.insert(ma[3][0]);
    se.insert(ma[2][1]);
    se.insert(ma[3][1]);
    if(se.size()!=4) return 0;
    se.clear();
    se.insert(ma[2][2]);
    se.insert(ma[3][2]);
    se.insert(ma[2][3]);
    se.insert(ma[3][3]);
    if(se.size()!=4) return 0;
    return 1;
}
int checkrow(int id)
{
    se.clear();
    int num = 0;
    for(int i  = 0 ;  i < 4; i++)
    {
        if(ma[id][i] == 0) num++;
        else
        se.insert(ma[id][i]);
    }
    return (se.size()+num) == 4;
}
int checkcol(int id)
{
    se.clear();
    int num = 0;
    for(int i  = 0 ;  i < 4; i++)
    {
        if(ma[i][id] == 0) num++;
        else
        se.insert(ma[i][id]);
    }
    return (se.size()+num) == 4;
}
int f;
void dfs(int id)
{
    if(id == cnt)
    {
        if(check()) f = 1;
        return ;
    }
    for(int i = 0; i < 4 && !f; i++)
    {
       ma[sx[id]][sy[id]] = a[i];
       if(checkcol(sy[id])&&checkrow(sx[id]))
       {
           //test();
           dfs(id+1);
       }
       if(f)break;
       ma[sx[id]][sy[id]] = 0;//一定不要忘啦！！！！！
    }
    return ;
}


int main()
{
   // ios::sync_with_stdio(0);
    int t;
    scanf("%d",&t);
    int cas=1;
    getchar();
    while(t--)
    {
        char s[10];
        cnt = 0;
        f = 0;
        for(int i  = 0 ;  i < 4; i++)
        {
             scanf("%s",s);
            for(int j = 0 ; j <4; j++)
            {
                if(s[j] =='*')
                {
                    ma[i][j] = 0;
                    sx[cnt] = i;
                    sy[cnt++] = j;
                }
                else
                {
                    ma[i][j] = s[j]-'0';
                }
            }
        }
        printf("Case #%d:\n",cas++);
        dfs(0);
        test();
    }
    return 0;
}