poj 3074 Sudoku

题意：解出9*9数独

思路：和poj 3076一样，只不过编码函数有点不同，依然使用舞蹈链

题目链接：http://poj.org/problem?id=3074

#include<cstdio>
#include<vector>
#include<cstring>
using namespace std;

const int maxn=400;
const int maxnode=3000;
const int maxr=1000;
struct DLX
{
    int n,sz;
    int s[maxn];
    int row[maxnode],col[maxnode];
    int r[maxnode],l[maxnode],u[maxnode],d[maxnode];
    int ansd,ans[maxr];

    void init(int n)
    {
        this->n=n;
        for(int i=0;i<=n;i++)
        {
            l[i]=i-1,r[i]=i+1,u[i]=d[i]=i;
        }
        l[0]=n,r[n]=0;
        sz=n+1;
        memset(s,0,sizeof(s));
    }

    void addrow(int R,const vector<int> &v)
    {
        int first=sz;
        for(int i=0;i<(int)v.size();i++)
        {
           int c=v[i];
           l[sz]=sz-1,r[sz]=sz+1,d[sz]=c,u[sz]=u[c];
           d[u[c]]=sz,u[c]=sz;
           row[sz]=R,col[sz]=c;
           s[c]++;sz++;
        }
        r[sz-1]=first;l[first]=sz-1;
    }

    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];--s[col[j]];
            }
        }
    }

    void Restore(int c)
    {
        for(int i=u[c];i!=c;i=u[i])
        {
            for(int j=l[i];j!=i;j=l[j])
            {
                ++s[col[j]];
                u[d[j]]=j,d[u[j]]=j;
            }
        }
        l[r[c]]=c;
        r[l[c]]=c;
    }

    bool dfs(int cur)
    {
        if(r[0]==0)
        {
            ansd=cur;
            return true;
        }

        int c=r[0];
        for(int i=r[0];i!=0;i=r[i])
            if(s[i]<s[c]) c=i;
        Remove(c);
        for(int i=d[c];i!=c;i=d[i])
        {
            ans[cur]=row[i];
            for(int j=r[i];j!=i;j=r[j]) Remove(col[j]);
            if(dfs(cur+1)) return true;
            for(int j=l[i];j!=i;j=l[j]) Restore(col[j]);
        }
        Restore(c);
        return false;
    }

    bool solve(vector<int> &v)
    {
        v.clear();
        if(!dfs(0)) return false;
        for(int i=0;i<ansd;i++) v.push_back(ans[i]);
        return true;
    }
};

DLX dlx;

const int slot=0;
const int row=1;
const int col=2;
const int sub=3;

int encode(int a,int b,int c)
{
    return a*81+b*9+c+1;
}

void decode(int code,int &a,int &b,int &c)
{
    code--;
    c=code%9,code/=9;
    b=code%9,code/=9;
    a=code;
}

char s[100],a[10][10];
int main(void)
{
    vector<int> v1;
    while(scanf("%s",s),s[0]!='e')
    {
        dlx.init(324);
        int cnt=0;
        for(int i=0;i<9;i++)
        {
            for(int j=0;j<9;j++)
                a[i][j]=s[cnt++];
            a[i][9]='\0';
        }
        for(int r=0;r<9;r++)
        {
            for(int c=0;c<9;c++)
            {
                for(int v=0;v<9;v++)
                {
                    if(a[r][c]=='.'||a[r][c]=='1'+v)
                    {
                        v1.clear();
                        v1.push_back(encode(slot,r,c));
                        v1.push_back(encode(row,r,v));
                        v1.push_back(encode(col,c,v));
                        v1.push_back(encode(sub,(r/3)*3+c/3,v));
                        dlx.addrow(encode(r,c,v),v1);
                    }
                }
            }
        }
        v1.clear();
        dlx.solve(v1);
        int r,c,v;
        for(int i=0;i<(int)v1.size();i++)
        {
            decode(v1[i],r,c,v);
            a[r][c]='1'+v;
        }
        for(int i=0;i<9;i++)
            printf("%s",a[i]);
        printf("\n");
    }
    return 0;
}