POJ 3067 Sudoku DLX

和POJ 3074一样

#include<cstdio> #define inf 1e8 #define M 16*16*16*16*16*4+100 #define N 16*16*16+100 int U[M],D[M],L[M],R[M],C[M],X[M];//x代表M所属的行，c代表M所属的列，U，D，L，R为一个元素的上下左右指针 int H[N],S[N],Q[N];//H是水平循环链表的头指针，S代表每一列的元素个数，Q存储一个可能结果 char map[20][20]; int size; void remove(int c) { R[L[c]]=R[c],L[R[c]]=L[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[C[j]]; } void resume(int c) { R[L[c]]=L[R[c]]=c; for(int i=U[c]; i!=c; i=U[i]) for(int j=L[i]; j!=i; j=L[j]) ++S[C[U[D[j]]=D[U[j]]=j]]; } bool Dance(int k) { int i,j,tmp,c; if(!R[0]) { for(i=0;i<k;i++) map[(X[Q[i]]/16)/16][(X[Q[i]]/16)%16]=X[Q[i]]%16+'A'; for(i=0;i<16;++i){ for(j=0;j<16;j++) printf("%c",map[i][j]); puts(""); } return 1; } for(tmp=inf,i=R[0];i;i=R[i]) if(S[i]<tmp)tmp=S[c=i]; remove(c); for(i=D[c]; i!=c; i=D[i]) { Q[k]=i; for(j=R[i]; j!=i; j=R[j])remove(C[j]); if(Dance(k+1))return 1; for(j=L[i]; j!=i; j=L[j])resume(C[j]);//在这里我遵循的原则是先删除的后还原，后删除的先还原。 } resume(c); return 0; } void Link(int r,int c) { ++S[C[size]=c]; D[size]=D[c]; U[D[c]]=size; U[size]=c; D[c]=size; if(H[r]<0)H[r]=L[size]=R[size]=size; else { R[size]=R[H[r]]; L[R[H[r]]]=size; L[size]=H[r]; R[H[r]]=size; } X[size++]=r; } void prepare(int m) { int i,j,num; for(i=0; i<=m; ++i) //注意列下标是从1开始到m. { S[i]=0; D[i]=U[i]=i; L[i+1]=i; R[i]=i+1; } R[m]=0; L[0]=m; size=m+1; for(i=0;i<=N;++i) H[i]=-1; } void cal(int &r,int &cx,int &cy,int &cg,int &ck,int i,int j,int k){ r=(i*16+j)*16+k-1; //第几行（行） cg=i*16+j+1; //第几个格子（列） 这里必须从1开始（我在这里WA了多次） cx=16*16+i*16+k; //第i行上的k（列） cy=16*16*2+j*16+k; //第j列上的k（列） ck=16*16*3+(i/4*4+j/4)*16+k;//第某个宫上的k（列） } int main(){ int i,j,k; int tx,ty; int r,cx,cy,cg,ck; while(scanf("%s",map[0])!=EOF){ for(i=1;i<16;i++) scanf("%s",map[i]); prepare(16*16*4); for(i=0;i<16;i++) for(k=0;k<16;k++){ if(map[i][k]=='-'){ for(j=1;j<=16;j++){ cal(r,cx,cy,cg,ck,i,k,j); Link(r,cx);Link(r,cy);Link(r,cg);Link(r,ck); } } else{ cal(r,cx,cy,cg,ck,i,k,map[i][k]-'A'+1); Link(r,cx);Link(r,cy);Link(r,cg);Link(r,ck); } } Dance(0); printf("\n"); } }