欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

搜索与图论——DFS与BFS

程序员文章站 2023-12-26 23:26:15
...
名称 数据结构 空间 特点
DFS深度优先搜索 staek O(h) 不具有最短路
BFS宽度优先搜索 queue O(2h) 最短路

2. DFS搜索顺序
深度优先搜索,每次总是搜索到一条路径上的最终子节点,然后开始回溯,搜索下一个分支节点。下图以三层树为例:搜索与图论——DFS与BFS
3. 例题1:全排列问题

给定一个从1开始长度为 n 的不包含重复数字的序列,请你求出其所有不重复的全排列。

输入格式
第一行包含整数 n。

输出格式
输出所有的不同排列,每种排列占一行。

在确定每种排列的输出顺序时,第一个数较小的先输出,第一个数相同时,第二个数较小的先输出,以此类推。

数据范围
1≤n≤9,
数组中包含的元素的取值范围 [1,9]
输入样例:
3
输出样例:
1 2 3
1 3 2
2 1 3
2 3 1
3 1 2
3 2 1

#include<iostream>

using namespace std;

const int N = 10;

int n;
int path[N];	// 存储每次搜索的路径,即每种情况对应的答案
bool st[N];		// 记录n个数是否被使用过,true为被使用过,false为未使用

// 函数功能:递归实现搜索
// 每一层递归相当于树的每一层,确定第i个空填几
void dfs(int u)
{
	// 递归终止条件:搜索到最后一层即最底层叶子节点
	if(u == n) // 如果递归的层数等于n则代表n个空全填满
	{
		for(int i = 0; i < n; i++) printf("%d ", path[i]);
		puts("");
		return;
	}

	// 判断该空应该填哪个未使用的数
	for(int i = 1; i <= n; i++)
		if(!st[i]) 
		{
			path[u] = i;
			st[i] = true;  // 标记已使用
			dfs(u + 1);		// 向下一层递归
			st[i] = false;	// 回溯,保留原状态
		}
}

int main()
{
	cin >> n;

	dfs(0);

	return 0;
}			

解题思路: 每一层递归在第i个空位中填一个未被使用的数,每次填到最后一个空位后开始回溯。

例题2:n皇后问题
n皇后问题是指将n个皇后放在n*n的国际象棋棋盘上,使得皇后不能相互攻击到,即任意两个皇后不能处于同一行、同一列或同一斜线上。
搜索与图论——DFS与BFS
现在给定整数n,请你输出所有的满足条件的棋子摆法。

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

输出格式
每个解决方案占n行,每行输出一个长度为n的字符串,用来表示完整的棋盘状态。
其中“.”表示一个位置的方格状态为空,“Q"表示某一个位置的方格上摆着皇后。
每个方案输出完成后,输出一个空行。

数据范围
1≤n≤9

输入示例:
4
输入示例:
…Q.
Q…
…Q
.Q…

.Q…
…Q
Q…
…Q.

解法一:以每一行递归的顺序搜索
#include<iostream>

using namespace std;

const int N = 20; 	// 开两倍大小,防止数组坐标为负

int n;
char g[N][N];	// 储存皇后的位置
bool col[N], dg[N], udg[N];	// 分别储存列,正对角线,反对角线上是否有皇后,true代表有,false代表没有

// 函数功能:递归求每一行皇后放置的位置
void dfs(int u)
{
	// 递归终止条件:搜索到最后一层即最底层叶子节点
	if(u == n) // 如果层数为n则输出
	{
		for(int i = 0; i < n; i++) puts(g[i]);
		puts("");
		return;
	}
	
	// 枚举每一列是否可以放皇后
	for(int i = 0; i < n; i++)
		if(!col[i] && !dg[u + i] && !udg[n - u + i])
		{
			g[u][i] = 'Q';
			col[i] = dg[u + i] = udg[n - u + i] = true;
			dfs(u + 1);
			col[i] = dg[u + i] = udg[n - u + i] = false;
			g[u][i] = '.';
		}
}

int main()
{
	cin >> n;
	for(int i = 0; i < n; i++)
		for(int j = 0; j < n; j ++)
			g[i][j] = '.';

	dfs(0);

	return 0;
}		

解题思路:因为每一行只能有一个皇后,所以只用判断每一层的皇后放在第几列就可以。这与例1的思路一样。

解法二:以每一个格子为顺序搜索
#include<iostream>

using namespace std;

const int N = 20; 	// 开两倍大小,防止数组坐标为负

int n;
char g[N][N];	// 储存皇后的位置
bool row[N], col[N], dg[N], udg[N];	// 分别储存列,正对角线,反对角线上是否有皇后,true代表有,false代表没有

// 函数功能:递归求每一行皇后放置的位置
void dfs(int x, int y, int s)
{
	if(y == n) y = 0, x++;

	if(x == n)
	{
		if(s == n)
		{
			for(int i = 0; i < n; i++) puts(g[i]);
			puts("");
		}
		return;
	}

	// 不放皇后
	if(!row[x] && !col[y] && !dg[x + y] && !udg[x - y + n])
	{
		g[x][y] = 'Q';
		row[x] = col[y] = dg[x + y] = udg[x - y + n] = true;
		dfs(x, y + 1, s+1);
		row[x] = col[y] = dg[x + y] = udg[x - y + n] = false;
		g[x][y] = '.';
	}			
}

int main()
{
	cin >> n;
	for(int i = 0; i < n; i++)
		for(int j = 0; j < n; j ++)
			g[i][j] = '.';

	dfs(0, 0, 0);

	return 0;
}	
相关标签: 算法 dfs

上一篇:

下一篇: