POJ3074 Sudoku 舞蹈链 DLX

欢迎访问~原文出处——博客园-zhouzhendong

去博客园看该题解

题目（传送门）

题意概括

给出一个残缺的数独，求解。

题解

DLX + 矩阵构建  （两个传送门）

代码

#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cstdlib>
#include <cmath>
using namespace std;
const int N=,M=,S=N*+M;
struct DLX{
    int n,m,cnt;
    int x[S],y[S],L[S],R[S],U[S],D[S];
    int C[M],anscnt,ans[N];
    void init(int c){
        memset(x,,sizeof x),memset(y,,sizeof y);
        memset(L,,sizeof L),memset(R,,sizeof R);
        memset(U,,sizeof U),memset(D,,sizeof D);
        memset(C,,sizeof C),memset(ans,,sizeof ans);
        anscnt=,m=c;
        for (int i=;i<=m;i++)
            L[i]=i-,R[i]=i+,U[i]=D[i]=i;
        L[]=m,R[m]=,cnt=m;
    }
    void link(int i,int j){
        cnt++;
        x[cnt]=i;
        y[cnt]=j;
        L[cnt]=cnt-;
        R[cnt]=cnt+;
        D[cnt]=j;
        D[U[j]]=cnt;
        U[cnt]=U[j];
        U[j]=cnt;
        C[j]++;
    }
    void Delete(int k){
        L[R[k]]=L[k];
        R[L[k]]=R[k];
        for (int i=D[k];i!=k;i=D[i])
            for (int j=R[i];j!=i;j=R[j]){
                U[D[j]]=U[j];
                D[U[j]]=D[j];
                C[y[j]]--;
            }
    }
    void Reset(int k){
        L[R[k]]=k;
        R[L[k]]=k;
        for (int i=U[k];i!=k;i=U[i])
            for (int j=L[i];j!=i;j=L[j]){
                U[D[j]]=j;
                D[U[j]]=j;
                C[y[j]]++;
            }
    }
    bool solve(){
        if (R[]==)
            return true;
        anscnt++;
        int k=R[];
        for (int i=R[k];i!=;i=R[i])
            if (C[i]<C[k])
                k=i;
        Delete(k);
        for (int i=D[k];i!=k;i=D[i]){
            ans[anscnt]=x[i];
            for (int j=R[i];j!=i;j=R[j])
                Delete(y[j]);
            if (solve())
                return true;
            for (int j=L[i];j!=i;j=L[j])
                Reset(y[j]);
        }
        Reset(k);
        anscnt--;
        return false;
    }
}dlx;
int a[][],x[N],y[N],z[N];
char s[];
int hash(int a,int b,int c){
    return a*+b*+c+;
}
int main(){
    while (scanf("%s",s)&&strlen(s)==){
        for (int i=;i<=;i++)
            for (int j=;j<=;j++){
                int pos=(i-)*+j-;
                if (s[pos]=='.')
                    a[i][j]=;
                else
                    a[i][j]=s[pos]-'';
            }
        dlx.init();
        int Row=;
        for (int i=;i<=;i++)
            for (int j=;j<=;j++){
                int st,en;
                if (a[i][j]==)
                    st=,en=;
                else
                    st=en=a[i][j];
                for (int k=st;k<=en;k++){
                    Row++;
                    x[Row]=i,y[Row]=j,z[Row]=k;
                    int first=dlx.cnt+;
                    dlx.link(Row,hash(,i-,j-));
                    dlx.link(Row,hash(,i-,k-));
                    dlx.link(Row,hash(,j-,k-));
                    dlx.link(Row,hash(,((i-)/)*+(j-)/,k-));
                    dlx.L[first]=dlx.cnt;
                    dlx.R[dlx.cnt]=first;
                }
            }
        bool found=dlx.solve();
        for (int i=;i<=dlx.anscnt;i++)
            a[x[dlx.ans[i]]][y[dlx.ans[i]]]=z[dlx.ans[i]];
        for (int i=;i<=;i++)
            for (int j=;j<=;j++)
                printf("%d",a[i][j]);
        puts("");
    }
    return ;
}