kuangbing题单(1)简单搜索POJ 1321 棋盘问题

先附上题目链接
一道简单的深搜.
两种方法(第二种比较慢但是感觉比较好理解)
状态设计上只需要得到当前的行数r和已经下过的棋子数i.
一个边界是行数全部搜索完,即r==n(数组从0-n-1).如果此时i == k.tot+1.
另一个边界是如果下过的棋子已经到达了k,tot+1.递归结束.
细节上面要注意遍历当前行数的列时,在循环里的条件是如果此格是棋盘位置并且没有被放置过棋子,标记后搜索i+1,r+1.但是如果不符合这个条件的话,不应该搜索i+1,r.而是在循环之后搜索i+1,r.
简单来说就是当前行不放棋子的话这种情况只要搜索一次.不能重复搜索.
第一种方法的代码:

#include <iostream>
#include <vector>
#include <string.h>
#include <algorithm>
using namespace std;
const int mn = 10;
const int INF = 0x3f3f3f3f;
int n, k;
char maze[mn][mn];
bool visc[mn];
long long tot;
void dfs(int r,int i) {
	if (i == k) {
		++tot;
		return;
	}
	if (r == n) {
		if (i == k) ++tot;
		return;
	}
	for (int j = 0; j < n; ++j) {
		if (visc[j]) continue;
		if (maze[r][j] == '#') {
			visc[j] = 1;
			dfs(r + 1, i + 1);
			visc[j] = 0;
		}
	}
	dfs(r + 1, i);
}
int main() {
	while (cin >> n >> k && n != -1 && k != -1) {
		for (int i = 0; i < n; ++i) {
			for (int j = 0; j < n; ++j) {
				cin >> maze[i][j];
			}
		}
		tot = 0;
		memset(visc, 0, sizeof(visc));
		if (k != 0)dfs(0, 0);
		cout << tot << endl;
	}

	return 0;
}

第二种方法其实很笨,但是比较理解.就是用一个vector(用数组也一样)存放’#‘的行列坐标(即棋盘位置).记录完之后只需要暴力的枚举每个’#'是否选还是不选(要注意的是此时vis数组应该开两个,一个记录行是否放过棋子,一个记录列是否放过棋子,如果放过的话是不可以选的即dfs只有一条分支).全部选择完之后进行判断下的棋子数是否等于k(其实中途在等于k的时候已经可以return了).
时间上第一种方法32ms第二种125ms.
附上第二种的code.

#include <iostream>
#include <vector>
#include <string.h>
#include <algorithm>
using namespace std;
const int mn = 10;
const int INF = 0x3f3f3f3f;
struct node {
	node(int a, int b) {
		this->r = a;
		this->c = b;
	}
	int r, c;
};
int n, k;
char maze[mn][mn];
bool visr[mn];
bool visc[mn];
long long tot;
vector<node> Node;
int len;
void dfs(int i, int cur) {
	if (cur == k) {
		++tot;
		return;
	}
	if (i == len) {
		if (cur == k) ++tot;
		return;
	}
	dfs(i + 1, cur);
	if (!visr[Node[i].r] && !visc[Node[i].c]) {
		visr[Node[i].r] = 1;
		visc[Node[i].c] = 1;
		dfs(i + 1, cur + 1);
		visr[Node[i].r] = 0;
		visc[Node[i].c] = 0;
	}
}
int main() {
	while (cin >> n >> k && n != -1 && k != -1) {
		Node.clear();
		for (int i = 0; i < n; ++i) {
			for (int j = 0; j < n; ++j) {
				cin >> maze[i][j];
				if (maze[i][j] == '#') {
					Node.push_back(node(i, j));
				}
			}
		}
		tot = 0;
		memset(visr, 0, sizeof(visr));
		memset(visc, 0, sizeof(visc));
		len = Node.size();
		dfs(0, 0);
		cout << tot << endl;
	}

	return 0;
}