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
=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;
}