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

DFS

程序员文章站 2022-07-12 10:29:09
...

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 树的过程中求出每一个节点为根节点的子树的节点数量。这是因为树的 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 号点。
}

参考:AcWing

相关标签: 算法