POJ 2676 数独（基础搜索）

题目：http://poj.org/problem?id=2676

题目大意：给你一个9*9的矩阵，里面又分为9个3*3的矩阵，有的格子里有一些1~9的数字，0表示没有数字，现在让你填1~9的数字，使得每一行每一列没个小矩阵的数字都是1~9这9个数字各出现一次。

解题思路：暴力搜索。用一个剪枝，用row[ i ][ j ] 表示第 i 行数字 j 有没有出现，col[ i ][ j ] 也一样，grid[ i ][ j ][ k ] 也是同样的道理，i、j用来定位，k表示数字k，然后直接暴力就好。

虽然说是基础，可是我第一遍竟然不敢这么做！果然几个月没做题，整个人都不好了。。

代码如下：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

int row[11][11],col[11][11],grid[11][11][11];
int ans[111];
int ok = 0;
char str[11][11];

void dfs(int x)
{
if(x == 81)
{
ok = 1;
return ;
}
if(str[x/9][x%9] != '0')
{
ans[x] = str[x/9][x%9]-'0';
dfs(x+1);
}
for(int i = 1;i <= 9;i++)
{
if(ok) return ;

if(row[x/9][i] == 1) continue;
if(col[x%9][i] == 1) continue;
if(grid[(x/9)/3][(x%9)/3][i] == 1) continue;
row[x/9][i] = 1;
col[x%9][i] = 1;
grid[(x/9)/3][(x%9)/3][i] = 1;
ans[x] = i;
dfs(x+1);
row[x/9][i] = 0;
col[x%9][i] = 0;
grid[(x/9)/3][(x%9)/3][i] = 0;
}
}

int main()
{
int _;
scanf("%d",&_);
while(_--)
{
memset(row,0,sizeof(row));
memset(col,0,sizeof(col));
memset(grid,0,sizeof(grid));
for(int i = 0;i < 9;i++)
{
scanf("%s",str[i]);
for(int j = 0;j < 9;j++)
{
row[i][str[i][j]-'0'] = 1;
col[j][str[i][j]-'0'] = 1;
grid[i/3][j/3][str[i][j]-'0'] = 1;
}
}
ok = 0;
dfs(0);
for(int i = 0;i < 81;i++)
{
if(i > 0 && i%9 == 0) puts("");
printf("%d",ans[i]);
//printf("i = %d",i);
}
puts("");
}
return 0;
}