数独
数独是一种传统益智游戏,你需要把一个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;
}