《算法竞赛进阶指南》数独

数独

数独是一种传统益智游戏，你需要把一个9 × 9的数独补充完整，使得图中每行、每列、每个3 × 3的九宫格内数字1~9均恰好出现一次。

请编写一个程序填写数独。

输入格式
输入包含多组测试用例。

每个测试用例占一行，包含81个字符，代表数独的81个格内数据（顺序总体由上到下，同行由左到右）。

每个字符都是一个数字（1-9）或一个”.”（表示尚未填充）。

您可以假设输入中的每个谜题都只有一个解决方案。

文件结尾处为包含单词“end”的单行，表示输入结束。

输出格式
每个测试用例，输出一行数据，代表填充完全后的数独。

输入样例：
4…8.5.3…7…2…6…8.4…1…6.3.7.5…2…1.4…
…52…8.4…3…9…5.1…6…2…7…3…6…1…7.4…3.
end
输出样例：
417369825632158947958724316825437169791586432346912758289643571573291684164875293
416837529982465371735129468571298643293746185864351297647913852359682714128574936

本题的整体思想是利用dfs，然后利用二进制标记的方法，col、row、cell分别存储每一列、行、3*3宫格中需要填充的数字，直接存储一个int值，它的二进制中的某一位为1的话就表示当前值需要填充。

#include <iostream>
#include <algorithm>
using namespace std;
const int N=9;
int col[N],row[N],cell[3][3];
//col用二进制压缩存储每一列存在的数字
//row同理存储每一行存在的数字
//cell存储每一个3*3存在的数字
int ones[1<<N],map[1<<N];
//ones数组存储每一个int值二进制中包含的1的数量
//map存储1、2、8……中二进制中1位于的位置
char str[100];
inline int lowbit(int x)
{
	return x& -x;
}
void init()
{
	for(int i=0;i<N;i++)
		row[i]=col[i]=(1<<N)-1;//使得保存的都是N位的1
	for(int i=0;i<3;i++)
		for(int j=0;j<3;j++)
			cell[i][j]=(1<<N)-1;
}
inline int get(int x,int y)
{
	return row[x]&col[y]&cell[x/3][y/3];
}
bool dfs(int cnt)
{
	if(!cnt)return true;//当所有的点都被填充时说明，当前已经确定了唯一的答案

	//在优化本题的步骤中，我们应最先遍历当前需要遍历的点中情况最少的，即和他有关的数最多的
	//这在dfs搜索的数中表现的是尽量在深度较小的位置进行剪枝
	int minv=10;//初始化最小值
    int x,y;//确定最小值位置的点
	for(int i=0;i<N;i++)
		for(int j=0;j<N;j++)
          if(str[i*9+j]=='.')
          {
          	int t=get(i,j);//获得当前点需要填充的int值
            if(ones[t]<minv)
            {//当当前需要填充的数字的个数小于最小值时，更新
            	minv=ones[t];
            	x=i,y=j;
            }
          }

   //接下来由确定的位置进行dfs填充
   for(int i=get(x,y);i;i-=lowbit(i))
   {//确定点当前位置需要填充的二进制值
   	 int t=map[lowbit(i)];//逐步确定需要填充数字的大小
       
       //对当前位置进行填充
       row[x]-=1<<t;
       col[y]-=1<<t;
       cell[x/3][y/3]-=1<<t;
       str[x*9+y]=t+'1';
       if(dfs(cnt-1)) return true;//每个测试样例只有唯一的答案

       row[x]+=1<<t;
       col[y]+=1<<t;
       cell[x/3][y/3]+=1<<t;
       str[x*9+y]='.';
   }     
   return false;
}
int main()
{
	for(int i=0;i<N;i++)map[1<<i]=i;
	for(int i=0;i<1<<N;i++)
	{
		int s=0;
		//lowbit返回当前二进制中首先遇到的值
		//即：若原值位1010100，返回的二进制为：100
		for(int j=i;j;j-=lowbit(j))s++;//每次取最先遇到的的二进制，这样就能循环遍历出所有的二进制中的1
		ones[i]=s;
	}
	while(cin>>str,str[0]!='e')
	{
		init();//首先初始话将col/row/cell数组所有的二进制为标记为1，表示目前都没有填上
	    
	    int cnt=0;//cnt记录需要填充的点的个数
        for(int i=0;i<N;i++)
        	for(int j=0;j<N;j++)
        		if(str[i*9+j]!='.')
        		{//当某一位置已经填充上数字时，在相关的行列以及三宫格中进行标记
                  int t=str[i*9+j]-'1';
                  row[i]-=1<<t;
                  col[j]-=1<<t;
                  cell[i/3][j/3]-=1<<t;
        		}
        		else cnt++;
      dfs(cnt);
      cout<<str<<endl;
	}
	return 0;
}