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");
	}
}