POJ1321 棋盘 DFS+回溯

嗯继续练DFS。发现DFS跟回溯好像比较好结合，要牢牢地掌握这种思想。

题目概述：

前人总结这道题说是简化版的八皇后问题，其实确实是这样，略有不同之处就是首先这个棋盘并不是完整意义上的 n x n 形状，而是不规则的形状，其可以放置棋子的地方都用 '#' 标注出来了。然后要求每一横行，每一纵行都至多有一个棋子，然后最后计算出可以这么摆放的形式的个数。

算法思想：

依旧是DFS+回溯，跟我之前写的木棒那篇的思想差不多（http://blog.csdn.net/oh233/article/details/42127443）。只是我现在更加理解了当时那道题的思想。

首先是开a[10][10]来记录棋盘形式，可以用1表示能放置棋子，0表示不能放置。然后另一个数组vis[10]表示的是这一纵行是否还可以防止棋子。

遍历方式是第一行到第n行这样。

在dfs函数中

首先注意第一个判断条件 (n-row+1) < k-num ，这里是说如果当前剩余的行数已经小于剩余的棋子数了，那直接退出，这个优化可以从47MS到0MS。

然后第二个条件，是说假如已经放置完毕了，那把res++，解法加一，再返回。

如果递归的次数已经达到n+1，也就是所有n行都完成但是没有到足够的k（似乎和第一个条件重复？），那就返回。

如果上面的条件都不对，那就进行回溯+递归，方法是检测这一行有没有可以放置的，如果有的话，就试探这个点可不可以，也就是设置vis[j]=1，然后运行下一行的dfs，如果不行就把vis[j]调回到1，如果这一行所有纵列都不满足条件，那就这一行不放，跳到下一行。

然后main函数直接调用，从第1行，当前0棋子放置来就好了。

代码部分：

#include <iostream>
#include <string.h>
using namespace std;
int a[10][10];
int vis[10];
int n, k;
int res;


void dfs(int row, int num){
	if (n - row + 1 < k - num) return;
	if (num == k) {
		res++;
		return;
	}
	if (row == n+1) return;
	for (int j = 1; j <= n; ++j){
		if (a[row][j] && !vis[j]) {
			vis[j] = 1;
			dfs(row + 1, num + 1);
			vis[j] = 0;
		}
	}
	dfs(row + 1, num);
}



int main() {
	while (cin >> n >> k && (n + k) != -2) {
		res = 0;
		memset(a, 0, sizeof(a));
		memset(vis, 0, sizeof(vis));
		for (int i = 1; i <= n; ++i){
			for (int j = 1; j <= n; ++j){
				char c; cin >> c;
				if (c == '#') a[i][j] = 1;
			}
		}
		dfs(1, 0);
		cout << res << endl;
	}
	return 0;
}