POJ 3074（DLX）

同样是数独的题，不同于poj 2676和2918，这题给的空白格很多，直接用dfs会超时，需要进一步的优化。

自己剪了半天的枝还是超时，搜了题解，发现了一种新的算法Dancing Link X（DLX），关于这个算法，推荐一篇博客https://www.cnblogs.com/grenet/p/3145800.html，这篇博客里写的很清楚了。

这样，我们就能把求解数独转换为精确覆盖问题。关键在于如何构造01矩阵。数独一共有四个约束条件

1.每个格子只能填一个数，1-81列便可定义为每个格子能否填数；

2.每行1-9都得填一遍，82-162列便可定义为每行的9种可能；

3.每列1-9都得填一遍，163-243同样代表每列的9种可能；

4.每一宫1-9得填一遍，244-324代表每一宫的9种可能；

这样，就能构造出729行，324列的01矩阵了。

然后直接套模板就能AC了

#include <cstdio>
#include <cstring>
#include <iostream>

using namespace std;
#define N 300000
#define INF 0x3fffffff
char g[10][10];
char s[100];
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;
            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()
{
    while(cin>>s)
    {
        if(s[0]=='e')
            break;
        build();
        int tu=0;
        int x = 1,y = 1;
        for(int i = 0; i < strlen(s); i++)
        {
            if(s[i]=='.')
                s[i] = '0';
            g[x][y] = s[i];
            if((i+1)%9==0)
            {
                x++;
                y = 1;
            }
            else
            {
                y++;
            }
        }


        for(int i=1;i<=9;i++)
        {
            for(int j=1;j<=9;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);
        for(int i = 1; i <= 9; i++)
        {
            for(int j = 1; j <= 9; j++)
            {
                cout<<g[i][j];
            }
        }
        cout<<endl;
    }
    return 0;
}