1321:棋盘问题

原创于 2022-03-17 17:11:08 发布 · 453 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#深度优先 #算法

poj 专栏收录该内容

65 篇文章

订阅专栏

本文介绍了使用DFS解决八皇后问题的思路和C++实现。通过从第一行开始放置棋子并递归搜索，同时设置列访问标记避免冲突，找到所有可能的解决方案。程序中详细展示了如何进行回溯以寻找不同的排列。

搜索，dfs
不难
思路：

从第0行开始，需要处理k个棋子，接下来是第一行，k-1个棋子，进行递归搜索（不用做行访问标记）
用vis[]数组，置列访问标记。
k=0时，退回到上一层递归，接着在for循环内寻找新的方案
直到退出第一层递归，遍历完成。

参考：https://blog.csdn.net/coding_sun/article/details/77412738

//搜索，dfs
#include <iostream>
#include <cstring>
using namespace std;

const int N = 8;

char chess[N][N];
int vis[N];
int n, k,cnt;

//从第a行开始，需要处理b个棋子
void dfs(int a, int b) {
	int i, j;	
	if (b == 0) {
		cnt++;
		return;//返回上层dfs,继续寻找下一种方案
	}
	for (i = a; i < n; i++) {
		for (j = 0; j < n; j++) {
			//假如chess[i][j]处是棋盘且该列未被访问过
			if (chess[i][j] == '#' && vis[j] == 0) {
				vis[j] = 1;
				dfs(i + 1, b - 1);//递归
				vis[j] = 0;//回溯
			}
		}
	}
}

int main() {	
	int i, j;
	while (cin >> n >> k) {
		if (n == -1) break;
		cnt = 0;
		memset(vis, 0, sizeof(vis));

		for (i = 0; i < n; i++) {
			for (j = 0; j < n; j++)
				cin >> chess[i][j];
		}
		dfs(0, k);
		cout << cnt<< endl;
	}
	return 0;
}