算法思维训练之DFS题过程详解及实战演练 (2月15日)

83 阅读2分钟

开启掘金成长之旅!这是我参与「掘金日新计划·2月更文挑战」的第14天,点击查看活动详情

题目链接:Click here ~~~

棋盘问题 Poj - 1321

在这里插入图片描述

1.棋盘问题题意及题解

给定一个棋盘,旗子(都相同)不能放在同一行或者同一列,明显,dfs入门模板题(类似于八皇后问题)。采用dfs算法,可以一行一行搜,也可以一列一列搜索。(以下给出一行一行搜索的方法)用一个二维数组char map[][]数组记录棋盘,用bool cln[] 布尔数组记录这一列是否有棋子,深搜结束条件:搜索到最后一行。判断条件:如果这一列没有棋子且这一行有棋子,就可以放一个棋子,然后继续搜索。(代码有注释,详情可看代码)


char map[10][10];//记录地图
bool cln[10];//查看这一列有没有棋子
int n, k;
int res;
void dfs(int r, int num) {      //深搜
	
	if (num == k) {         //结束条件
		res++;
		return;
	}
	if (r > n) return;
	for (int i = 1; i <= n; ++i) {
		if (cln[i] == false && map[r][i] == '#') { //判断条件,如果这一列没有棋子且这一行有棋子,就可以执行
			cln[i] = true;
			dfs(r + 1, num + 1);//搜索下一行
			cln[i] = false;
		}
	}
	dfs(r + 1, num);
}
int main() {
	while (cin >> n >> k) {
		if (n == -1 && k == -1) break;
		memset(cln, false, sizeof(cln));          //初始化cln为false即没有棋子
		for (int i = 1; i <= n; ++i)
			for (int j = 1; j <= n; ++j)
				cin >> map[i][j];//输入棋子
		dfs(1, 0);         //开始搜索
		printf("%d\n", res);
		res = 0;           //初始化结果,以免妨碍下一轮
	}
	return 0;
}

类似题目:

DFS处理全排列的问题

给定一个整数n,将数字1~n排成一排,将会有很多种排列方法。

现在,请你按照字典序将所有的排列方法输出。

输入格式 共一行,包含一个整数n。

输出格式 按字典序输出所有排列方案,每个方案占一行。

数据范围 1n71≤n≤7 输入样例:

3

输出样例:

1 2 3
1 3 2
2 1 3
2 3 1
3 1 2
3 2 1

在这里插入图片描述

DFS的路线图: 在这里插入图片描述

我们不需要存这个树,每次深搜走到底就是一条路线。

回溯的过程中需要注意:

恢复原样。flag[i] = false;

代码(爆搜):

1.C++代码实现

const int N = 10;
int n;
int path[N];      //保存路径
bool flag[N];
void dfs(int u) {         //u代表层数,u==n的时候就是叶节点。
	if (u == n) {
		for (int i = 0; i < n; i++) printf("%d ", path[i]);
		puts("");  //输出空行
		return;
	}
	for (int i = 1; i <= n; i++) {
		
		if (!flag[i]) {               //如果当前这个数没有被用过
			path[u] = i;          //填写i
			flag[i] = true;
			dfs(u + 1);           //进入下一层递归,递归函数之后记得要恢复原状
			flag[i] = false;      //回溯之后需要恢复原样
		}
	}
}
int main() {
	cin >> n;
	dfs(0);    //从0开始搜
	return 0;
}

经典八皇后问题

在这里插入图片描述

剪枝:提前终止搜索,直接回溯,不用继续往下搜索了。

定义:

char path[N][N]; //代表路径

bool col[N], dg[N], udg[N];//col代表列,dg代表正对角线,udg代表反对角线

1.第一种搜索方式 :按行搜索(全排列思想):

void dfs(int row) {
	if (row == n) {    //当u==n时,表示已经搜了n行,所以输出这条路径
		for (int i = 0; i < n; i++) cout << path[i] << endl;
		cout << endl;
		return;
	}

	
	for (int i = 0; i < n; i++)        //当前枚举第u行,看皇后放在哪一列
		
		if (!col[i] && !dg[row + i] && !udg[n - row + i]) {     // 剪枝,如果皇后在一列且正反对角线都没有。
			path[row][i] = 'Q';           //填写皇后
			col[i] = dg[row + i] = udg[n - row + i] = true;
			dfs(row + 1);
                        
			//回溯之后需要恢复原样
			col[i] = dg[row + i] = udg[n - row + i] = false;
			path[row][i] = '.';

		}
}
int main() {
	cin >> n;
	for (int i = 0; i < n; i++)
		for (int j = 0; j < n; j++)
			path[i][j] = '.';                 //地图上先打满点
	dfs(0);
	return 0;
}


解释一下dg和udg数组下标问题。 👆👆👆👆👆👆👆👆👆👆👆👆👆👆👆👆 在这里插入图片描述

2.第二种搜索方式 :(搜索每个点)

void dfs(int x, int y, int s) {
	if (s > n) return;
	if (y == n) {     //搜索到出界,换行
		y = 0;
		x++;
	}
	if (x == n) {
		if (s == n) {
			for (int i = 0; i < n; i++) cout << path[i] << endl;
			cout << endl;
			return;
		}
	}
	path[x][y] = '.';
	dfs(x, y + 1, s);
	//如果这一行没有,这一列没有,正对角线没有,反对角线也没有。则放一个皇后。
	if (!row[x] && !col[y] && !dg[x + y] && !udg[x - y + n]) {
		row[x] = col[y] = dg[x + y] = udg[x - y + n] = true;
		path[x][y] = 'Q';
		dfs(x, y + 1, s + 1);
		path[x][y] = '.';
		row[x] = col[y] = dg[x + y] = udg[x - y + n] = false;
	}
}
int main()
{
	cin >> n;
	dfs(0, 0, 0);
	return 0;
}