算法竞赛——进阶指南——acwing166. 数独 搜索

经典搜索题。

搜索加剪枝。

每次从能填数最少的空处开始填。

如果填了某个数，导致有空位不能填，则剪去

用二进制优化一下常数

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define ls (o<<1)
#define rs (o<<1|1)
#define pb push_back
const double PI= acos(-1.0);
const int M = 1e5+7;
/*
int head[M],cnt;
void init(){cnt=0,memset(head,0,sizeof(head));}
struct EDGE{int to,nxt,val;}ee[M*2];
void add(int x,int y){ee[++cnt].nxt=head[x],ee[cnt].to=y,head[x]=cnt;}
*/

char S[110];
char s[10][10];
bool f;
int sz;
int r[10],c[10],q[10],nm[600],val[600];
int lowbit(int x)
{
	return x&(-x);
}
int id(int x,int y)
{
	return (x-1)/3*3+(y-1)/3+1;;
}
void up(int x,int y,int z)
{
	r[x]^=(1<<(z-1));
	c[y]^=(1<<(z-1));
	q[id(x,y)]^=(1<<(z-1));
}
void dfs(int cnt)
{
	if(f)return ;//已经找到可行解了 
	if(cnt==82)//填完所有的空位 
	{
	//puts("oik");
		for(int i=1;i<=9;i++)for(int j=1;j<=9;j++)printf("%c",s[i][j]);
		puts("");
		f=true;
		return ;
	}
	int mn=(1<<9)-1,x,y;
	for(int i=1;i<=9;i++)
	for(int j=1;j<=9;j++)
	{
		if(s[i][j]!='.')continue;
		int tp=r[i]&c[j]&q[id(i,j)];
		if(!tp)return ;//存在一个位置，无数可填，剪去
		if(nm[tp]<nm[mn])mn=tp,x=i,y=j;
	}
	while(mn)
	{
		int z=val[lowbit(mn)];
		s[x][y]=('0'+z);up(x,y,z);
		dfs(cnt+1);
		s[x][y]='.';up(x,y,z);
		mn-=lowbit(mn);
	}
	return ;
}
int main()
{
	for(int i=0;i<(1<<9);i++)
	{
		int tp=i;
		while(tp)tp-=lowbit(tp),nm[i]++;
	}
	for(int i=1;i<=9;i++)val[(1<<(i-1))]=i;
	while(scanf("%s",S+1))
	{
		if(S[1]=='e')return 0;
		sz=0;f=false;
		for(int i=1;i<=9;i++)r[i]=c[i]=q[i]=(1<<9)-1;
		for(int i=1;i<=9;i++)
			for(int j=1;j<=9;j++)
			{
				s[i][j]=S[(i-1)*9+j];
				if(s[i][j]!='.')
				{
					sz++;
					up(i,j,s[i][j]-'0');
				}
			}
		dfs(sz+1);
	}
	return 0;
}