数独问题,今天蓝桥杯校赛没把这题做出来,和学长聊了一下,可能是我对于回溯的理解不够深入。所以今晚补了一下数独的代码。
解析:这题和八皇后问题相似,可以开一个3数组,分别标记该行用了什么数字,该列用了什么数字,和该方块用了什么数字,如果遇到标记过了数字,就跳过。
总结:这是省赛前的最后一题,复习一下回溯,祝愿我明天省赛好运。
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <string>
using namespace std;
const int N = 10;
char s[N][N][N];
int num[N][N];
bool ok;
int visx[N][N],visy[N][N],visz[N][N][N];
void dfs(int x,int y) {
if(y == 9) {
dfs(x+1,0);
return ;
}
if(x == 9) {
ok = true;
return ;
}
if(num[x][y] > 0) {
dfs(x,y+1);
return ;
}else {
for(int i = 1; i <= 9; i++) {
if(!visx[x][i] && !visy[y][i] && !visz[x/3][y/3][i]) {
visx[x][i] = visy[y][i] = visz[x/3][y/3][i] = true;
num[x][y] = i;
dfs(x,y+1);
if(ok) {
return ;
}
visx[x][i] = visy[y][i] = visz[x/3][y/3][i] = false;
num[x][y] = 0;
}
}
}
}
int main() {
int cas = 0 , tmp;
while(scanf("%s",s[0][0]) != EOF) {
if(cas++) {
printf("\n");
}
if(s[0][0][0] >= '1' && s[0][0][0] <= '9') {
num[0][0] = s[0][0][0] - '0';
tmp = num[0][0];
visx[0][tmp] = visy[0][tmp] = visz[0][0][tmp] = true;
}else {
num[0][0] = 0;
}
int j = 1;
for(int i = 0; i < 9; i++) {
for(; j < 9; j++) {
scanf("%s",s[i][j]);
if(s[i][j][0] >= '1' && s[i][j][0] <= '9') {
num[i][j] = s[i][j][0] - '0';
tmp = num[i][j];
visx[i][tmp] = visy[j][tmp] = visz[i/3][j/3][tmp] = true;
}else {
num[i][j] = 0;
}
}
if(j == 9) {
j = 0;
}
}
ok = false;
dfs(0,0);
for(int i = 0; i < 9; i++) {
for(int j = 0; j < 9; j++) {
printf("%d",num[i][j]);
if(j < 8) {
printf(" ");
}
}
printf("\n");
}
memset(s,0,sizeof(s));
memset(num,0,sizeof(num));
memset(visx,0,sizeof(visx));
memset(visy,0,sizeof(visy));
memset(visz,0,sizeof(visz));
}
return 0;
}