DFS
DFS 与 BFS 时间复杂度相同,都是 O ( n + m ) O(n + m) O(n+m),其中 n 表示图的节点数,m 表示图的边数。
相比于 BFS,DFS 代码量比 BFS 短很多,但是缺点是有可能 爆栈。从实际应用来看,能用 dfs 解决,就不要用 bfs 了。
DFS 会遍历所有的节点,但往往 BFS 不需要这样。
树与图的 DFS
树(不是图)的 DFS 有一个 特点:可以在 DFS 树的过程中求出每一个节点为根节点的子树的节点数量。这是因为树的 DFS 不会深搜上面的点,只能搜索下方的点,又因为是树而不是图,所以不同子树之间没有连接。
#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;
const int N = 1000010, M = 2 * N;
int h[N], e[M], ne[M], idx;
bool st[N];
void init() {
memset(h, -1, sizeof h);
}
void add(int a, int b) {
e[idx] = b;
ne[idx] = h[a];
h[a] = idx++;
}
void dfs(int u) {
st[u] = true; // dfs、bfs一般都需要标记一下,防止重复搜索同一个点
for (int i = h[u]; ~i; i = ne[i]) {
int j = e[i];
if (!st[j]) dfs(j);
}
}
int main()
{
init();
int n;
scanf("%d", &n);
int a, b;
for (int i = 1; i <= n - 1; ++i) {
scanf("%d%d", &a, &b);
add(a, b);
add(b, a); // 无向图
}
dfs(1); // 随便选择一个点作为遍历的起点,这里选择了第 1 号点。
}