POJ 2676 Sudoku (DFS回溯剪枝）

POJ 2676

设置三个标记数组，分别是行的，列的和小正方型的。  行列比较好处理，至于小正方形，我刚开始是直接把每个格子遍历一遍，然后算出他们分别在哪一个小正方形内，最后存在一个数组里。这个方法虽然比较麻烦但是很好想，也不会花太多时间。做完之后学长又说了另一个好一些的办法。每个小正方形的编号k（1~9）恰好就等于(i - 1) / 3 * 3 + (j - 1) / 3 + 1。 有了这个式子就简单多了。（关于这个式子的推导见参考博客，感谢！）然后就是普通的深搜加回溯。由于只需要得到一个满足的结果，所以我设置了一个flag来进行剪枝。不知道这个办法好不好，还请多指教。

参考博客：http://blog.csdn.net/lyy289065406/article/details/6647977   感谢！！！

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int map[15][15];
int visrow[15][15];
int viscol[15][15];
int vissq[15][15];
int flag = 0;
void dfs(int x, int y, int num0) {
	if(flag) return;			//剪枝，找出了解就直接退出即可 
	int i, j;
	if(num0 == 0 && !flag) {
		flag = 1;
		for(i = 1; i <= 9; i++) {
			for(j = 1; j <= 9; j++) {
				printf("%d", map[i][j]);
			}
			printf("\n");
		}
		return;
	}
	if(map[x][y] == 0) {
		for(i = 1; i <= 9; i++) {
			if(visrow[x][i] == 0 && viscol[y][i] == 0 && vissq[(x-1)/3*3+(y-1)/3+1][i] == 0) {
				map[x][y] = i;
				visrow[x][i] = 1;
				viscol[y][i] = 1;
				vissq[(x-1)/3*3+(y-1)/3+1][i] = 1;
				if(y != 9) dfs(x, y + 1, num0 - 1);
				else dfs(x + 1, 1, num0 - 1);
				map[x][y] = 0;
				visrow[x][i] = 0;
				viscol[y][i] = 0;
				vissq[(x-1)/3*3+(y-1)/3+1][i] = 0;
			}
		}
	}
	else {
		if(y != 9) dfs(x, y + 1, num0);
		else dfs(x + 1, 1, num0);
	}
}
int main () {
	//freopen("input.txt", "r", stdin);
	//freopen("output.txt", "w", stdout);
	int t;
	scanf("%d", &t);
	getchar();
	while(t--) {
		int i, j, num0 = 0;
		char ch;
		int n;
		for(i = 1; i <= 9; i++) {
			for(j = 1; j <= 9; j++) {
				visrow[i][j] = 0;
				viscol[i][j] = 0;
				vissq[i][j] = 0;
			}
		}
		for(i = 1; i <= 9; i++) {
			for(j = 1; j <= 9; j++) {
				scanf("%c", &ch);
				n = ch - '0' ;
				map[i][j] = n;
				if(map[i][j] == 0) num0++;
				visrow[i][map[i][j]] = 1;
				viscol[j][map[i][j]] = 1;
				vissq[(i-1)/3*3+(j-1)/3 + 1][map[i][j]] = 1;
			}
			getchar();
		}
		flag = 0;
		dfs(1, 1, num0);
	}
	return 0;
}