【模板】Dancing-Link X

模板題

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
#define N 1005
#define INF 0x3f3f3f3f
int n,m;
int row[N*N],s1[N];
int L[N*N],R[N*N],U[N*N],D[N*N];
int C[N*N],O[N];
int mp[1005][1005];
int head,dep;
int ans[85];
inline void remove(const 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];
			s1[C[j]]--;	
		}
}
void resume(const int& c)
{
	for(int i=U[c];i!=c;i=U[i])
		for(int j=L[i];j!=i;j=L[j])
		{
			s1[C[j]]++;
			U[D[j]]=j;
			D[U[j]]=j;
		}
	L[R[c]]=c;
	R[L[c]]=c;
}
inline bool dfs()
{
	if(R[head]==head)
		return 1;
	int min=INF,c;
	for(int i=R[head];i!=head;i=R[i])
		if(s1[i]<min)
		{
			min=s1[i];
			c=i;
		}
	remove(c);
	for(int i=D[c];i!=c;i=D[i])
	{
		O[dep++]=row[i];
		for(int j=R[i];j!=i;j=R[j])
			remove(C[j]);
		if(dfs())
			return 1;
		for(int j=L[i];j!=i;j=L[j])
			resume(C[j]);
		dep--;
	}
	resume(c);
	return 0;
}
char s[105];
int main()
{
	while(~scanf("%s",s+1))
	{
		memset(mp,0,sizeof(mp));
		if(s[1]=='e')
			return 0;
		m=4*81,n=9*81;
		for(int i=0;i<=m;i++)
			s1[i]=0,L[i]=i-1,R[i]=i+1,U[i]=D[i]=i;
        for(int i=1;i<=9;i++)
			for(int j=1;j<=9;j++)
			{
                int t=9*(i-1)+j;
                if(s[t]=='.')
					for(int k=1;k<=9;k++)
					{
						mp[9*(t-1)+k][t]=1;
						mp[9*(t-1)+k][81+(i-1)*9+k]=1;
						mp[9*(t-1)+k][162+(j-1)*9+k]=1;
						mp[9*(t-1)+k][243+(3*((i-1)/3)+(int)ceil(1.0*j/3)-1)*9+k]=1;
					}
               else
               {
					int k=s[t]-'0';
					mp[9*(t-1)+k][t]=1;
					mp[9*(t-1)+k][81+(i-1)*9+k]=1;
					mp[9*(t-1)+k][162+(j-1)*9+k]=1;
					mp[9*(t-1)+k][243+(3*((i-1)/3)+(int)ceil(1.0*j/3)-1)*9+k]=1;
               }
            }
		L[0]=m;
		R[m]=0;
		int siz=m+1;
		for(int i=1;i<=n;i++)
		{
			int bg=siz,ed=siz;
			for(int j=1;j<=m;j++)
			{
				int x=j;
				if(mp[i][j]==0)
					continue;
				s1[x]++;
				C[siz]=x;
				row[siz]=i;
				U[D[x]]=siz;
				D[siz]=D[x];
				U[siz]=x;
				D[x]=siz;
				L[siz]=ed;
				R[ed]=siz;
				R[siz]=bg;
				L[bg]=siz;
				ed=siz;
				siz++;
			}
		}
		dep=0;
		dfs();
		for(int i=0;i<dep;i++)
			ans[(O[i]-1)/9+1]=(O[i]%9==0)?9:O[i]%9;
		for(int i=1;i<=81;i++)
			printf("%d",ans[i]);
		printf("\n");
	}
	return 0;
}