HDU 1426 dancing links解决数独问题

 题目大意：

这是一个最简单的数独填充题目，题目保证只能产生一种数独，所以这里的初始9宫格较为稠密，可以直接dfs也没有问题

但最近练习dancing links，这类数据结构解决数独无疑效率会高很多

dancing links的数独限制条件是：

1.每行有9个元素，共9行 对应dlx81列

2.每列有9个元素，共9行 对应dlx81列

3.每个九宫格有9个元素，共9行 对应dlx81列

4.81个格子，每个格子最多有一个数

 

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <climits>
#include <cmath>

using namespace std;
#define N 1000
#define MAXNODE 1000000
const int INF = INT_MAX;
const double eps = 1e-8;

int a[10][10];
char str[5];

int belong[9][9] =  {
                        {1,1,1,2,2,2,3,3,3},
                        {1,1,1,2,2,2,3,3,3},
                        {1,1,1,2,2,2,3,3,3},
                        {4,4,4,5,5,5,6,6,6},
                        {4,4,4,5,5,5,6,6,6},
                        {4,4,4,5,5,5,6,6,6},
                        {7,7,7,8,8,8,9,9,9},
                        {7,7,7,8,8,8,9,9,9},
                        {7,7,7,8,8,8,9,9,9},
                    };
struct DLX{
    int n ,m , size;
    int col[MAXNODE] , row[MAXNODE];
    int U[MAXNODE] , D[MAXNODE] , L[MAXNODE] , R[MAXNODE];
    int cnt_col[N] , first[N];
    int ans[100];

    void init(int _n , int _m)
    {
        n = _n , m = _m;
        size= m ;
        for(int i=0 ; i<=m ; i++){
            L[i] = i-1 , R[i] = i+1;
            U[i] = D[i] = i;
        }
        L[0] = m , R[m] = 0;
        for(int i=1 ; i<=m ; i++) cnt_col[i] = 0;
        for(int i=1 ; i<=n ; i++) first[i] = -1;
    }

    void link(int r , int c)
    {
        ++size;
        U[D[c]] = size , D[size] = D[c];
        U[size] = c , D[c] = size;

        if(first[r]<0) L[size]=R[size]=first[r] = size;
        else{
            L[R[first[r]]] = size , R[size] = R[first[r]];
            L[size] = first[r] , R[first[r]] = size;
        }
        row[size] = r , col[size] = c , cnt_col[c]++;
    }

    void Remove(int c)
    {
        L[R[c]] = L[c] , R[L[c]] = R[c];
        for(int i=D[c] ; i!=c ; i=D[i]){
            for(int j=R[i] ; j!=i ; j=R[j]){
                U[D[j]] = U[j] , D[U[j]] = D[j];
                cnt_col[col[j]]--;
            }
        }
    }

    void Resume(int c)
    {
        for(int i=U[c] ; i!=c ; i=U[i]){
            for(int j=L[i] ; j!=i ; j=L[j]){
                U[D[j]] = D[U[j]] = j;
                cnt_col[col[j]]++;
            }
        }
        L[R[c]] = R[L[c]] = c;
    }

    bool Dance(int d)
    {
        if(!R[0]){
            for(int i=0 ; i<d ; i++){
                    int r = (ans[i]-1)/81;
                    int c = ((ans[i]-1)%81)/9;
                    a[r][c] = ((ans[i]-1)%9)+1;
            }
            return true;
        }
        int st=R[0];
        for(int i=R[0] ; i!=0 ; i=R[i])
            if(cnt_col[i]<cnt_col[st])
                st = i;
        Remove(st);
        for(int i=D[st] ; i!=st ; i=D[i]){
            ans[d] = row[i];
            for(int j=R[i] ; j!=i ; j=R[j]) Remove(col[j]);
            if(Dance(d+1)) return true;
            for(int j=L[i] ; j!=i ; j=L[j]) Resume(col[j]);
        }
        Resume(st);
        return false;
    }

}dlx;

void printM()
{
    for(int i=0 ; i<9 ; i++)
        for(int j=0 ; j<9 ; j++){
           if(j<8) printf("%d " , a[i][j]);
           else printf("%d\n" , a[i][j]);
        }
}

int main()
{
  //  freopen("a.in" , "r" , stdin);
    int index=0;
    bool flag = false;
    while(~scanf("%s" , str))
    {
        if(flag) puts("");
        flag = true;
        if(str[0] == '?') a[index/9][index%9] = 0;
        else a[index/9][index%9] = str[0]-'0';
        index++;
        while(index<81){
            scanf("%s" , str);
            if(str[0] == '?') a[index/9][index%9] = 0;
            else a[index/9][index%9] = str[0]-'0';
            index++;
        }
       // printM();
        dlx.init(729 , 324);
        for(int i=0 ; i<9 ; i++)
            for(int j=0 ; j<9 ; j++)
            {
                if(a[i][j]){
                    dlx.link((i*9+j)*9+a[i][j] , i*9+a[i][j]);
                    dlx.link((i*9+j)*9+a[i][j] , 81+j*9+a[i][j]);
                    dlx.link((i*9+j)*9+a[i][j] , 162+(belong[i][j]-1)*9+a[i][j]);
                    dlx.link((i*9+j)*9+a[i][j] , 243+i*9+j+1);
                }
                else{
                    for(int k=1 ; k<=9 ; k++){
                        dlx.link((i*9+j)*9+k , i*9+k);
                        dlx.link((i*9+j)*9+k , 81+j*9+k);
                        dlx.link((i*9+j)*9+k , 162+(belong[i][j]-1)*9+k);
                        dlx.link((i*9+j)*9+k , 243+i*9+j+1);
                    }
                }
            }
        dlx.Dance(0);

        printM();
        index=0;
    }

    return 0;
}