DLX 求解数独 模 板 NYOJ 722 , poj 2676

// 复杂的数独比DFS快//用空间来换时间#include <stdio.h>#include <string.h>#include <iostream>using namespace std;#define N 300000#define INF 0x3fffffffchar g[10][10];int ans[1000];int u[5000],d[5000],r[5000],l[5000],num[5000],H[1000],save[5000],save1[5000];int flag,head;const int n=729;const int m=324;int id;void prepare(){    for(int i=0;i<=m;i++)    {        num[i]=0;        d[i]=i;        u[i]=i;        r[i]=i+1;        l[i+1]=i;    }    r[m]=0;    memset(H,-1,sizeof(H)); // 记录每一行的第一个点}void link(int tn,int tm){    id++;    save1[id]=tn; // 记录行    ++num[save[id]=tm]; // 记录列    d[id]=d[tm];    u[ d[tm] ]=id;    u[id]=tm;    d[tm]=id;    if(H[tn]<0) H[tn]=l[id]=r[id]=id;    else    {        r[id]=r[H[tn]];        l[ r[H[tn]] ]=id;        r[ H[tn] ]=id;        l[id]=H[tn];    }}void build(){    id=m;    int sum;    prepare();    int tn=0;    for(int i=1;i<=81;i++)    {        for(int j=1;j<=9;j++)        {            ++tn;            link(tn,i);        }    }    sum=81;    /////////////////    for(int i=1;i<=9;i++) // 每一行    {        tn=(i-1)*81;        for(int k=1;k<=9;k++)        {            int tk=tn+k;            for(int j=1;j<=9;j++)            {                link(tk,sum+(i-1)*9+k);                tk+=9;            }        }    }    sum+=81;    ///////////////////////    for(int i=1;i<=9;i++)    {        tn=(i-1)*9;        for(int k=1;k<=9;k++)        {            int tk=tn+k;            for(int j=1;j<=9;j++)            {                link(tk,sum+(i-1)*9+k);                tk+=81;            }        }    }    sum+=81;    /////////////////////////    int tt=0;    for(int i1=1;i1<=3;i1++)    {        for(int j1=1;j1<=3;j1++)        {            tn=(i1-1)*81*3+9*3*(j1-1);            for(int k=1;k<=9;k++)            {                ++tt;                int tk;                for(int i=1;i<=3;i++)                {                    for(int j=1;j<=3;j++)                    {                        tk=tn+(i-1)*81+9*(j-1)+k;                        link(tk,sum+tt);                    }                }            }        }    }}void remove(int s){    l[ r[s] ]=l[s];    r[ l[s] ]=r[s];    for(int i=d[s];i!=s;i=d[i])        for(int j=r[i];j!=i;j=r[j])        {            u[d[j]]=u[j];            d[u[j]]=d[j];            num[save[j]]--;        }}void resume(int s){    r[l[s]]=s;    l[r[s]]=s;    for(int i=u[s];i!=s;i=u[i])        for(int j=l[i];j!=i;j=l[j])        {            u[d[j]]=j;            d[u[j]]=j;            num[save[j]]++;        }}void dfs(int s){    if(flag) return ;    if(r[head]==head)    {        flag=1;        for(int i=0;i<s;i++)        {            int ti,tj,tk;            int tans=save1[ans[i]]-1;            ti= (tans)/81+1;            tj= (tans%81)/9+1;;            tk= (tans%81)%9+1;            //printf("<%d %d> ",ti,tj);            g[ti][tj]=tk+'0';        }        return ;    }    int mi=INF,tu;    for(int i=r[head];i!=head;i=r[i])        if(mi>num[i])        {            mi=num[i];            tu=i;        }    remove(tu);    for(int i=d[tu];i!=tu;i=d[i])    {        for(int j=r[i];j!=i;j=r[j])            remove(save[j]);        ans[s]=i;        dfs(s+1);        for(int j=l[i];j!=i;j=l[j])            resume(save[j]);    }    resume(tu);}int main(){    int T;    scanf("%d",&T);    while(T--)    {        build();        int tu=0;        for(int i=1;i<=9;i++)        {            for(int j=1;j<=9;j++)            {                cin>>g[i][j];                if(g[i][j]!='0')                {                    int kk=g[i][j]-'0';                    remove( save[ H[tu+kk] ] );                    for(int i1=r[ H[tu+kk] ];i1 != H[tu+kk];i1=r[i1])                    {                        remove( save[i1] );                    }                }                tu+=9;            }        }        flag=0;        dfs(0);        printf("\n");        for(int i=1;i<=9;i++)        {            for(int j=1;j<=9;j++)                printf("%c ",g[i][j]);            printf("\n");        }    }    return 0;}