深度优先搜索（dfs）模版题

深度优先搜索理论基础 | 代码随想录

1. dfs和bfs区别

   1. 深度优先搜索（dfs）可一个方向去搜，不到黄河不回头，直到遇到绝境了，搜不下去了，再换方向

   2. 广度优先搜索（bfs）是先把本节点所连接的所有节点遍历一遍，走到下一个节点的时候，再把连接节点的所有节点遍历一遍，搜索方向更像是广度，四面八方的搜索过程。

2. dfs

例子：遍历从1-6的路径

1-5-6，1-5-4-6，1-5-4-3-6，1-5-4-2-3-6，节点5遍历完毕

1-4-6，1-4-3-6，1-4-2-3-6，节点4遍历完毕

1-2-3-6，1-2-4-6，1-2-4-5-6，节点2遍历完毕

换方向是撤销原路径，改为节点链接的下一个路径，回溯的过程

代码模版

void dfs(参数) {
    处理节点
    dfs(图，选择的节点); // 递归
    回溯，撤销处理结果
}

回溯框架

void backtracking(参数) {
    if (终止条件) {
        存放结果;
        return;
    }
    for (选择：本层集合中元素（树中节点孩子的数量就是集合的大小）) {
        处理节点;
        backtracking(路径，选择列表); // 递归
        回溯，撤销处理结果
    }
}

整体框架

void dfs(参数) {
    if (终止条件) {
        存放结果;
        return;
    }

    for (选择：本节点所连接的其他节点) {
        处理节点;
        dfs(图，选择的节点); // 递归
        回溯，撤销处理结果
    }
}

书写步骤：

• 确认递归函数，参数

一般情况下，需要二维数组保存所有路径，需要一维数组保存每次遍历的单一路径，为了书写方便可以定义一个全局变量

vector<vector<int>> result; // 保存符合条件的所有路径

vector<int> path; // 起点到终点的路径

void dfs (图，目前搜索的节点)

• 确认终止条件

非常重要，如果没有考清楚会出现 死循环，栈溢出等问题

if (终止条件) {

存放结果;

return;

}

• 处理当前节点出发的所有路径

for循环遍历

for (本节点所连接的其他节点) {

处理当前节点;

dfs(图，选择的节点); // 递归

回溯;

}

例题：

98. 所有可达路径

给定一个有 n 个节点的有向无环图，节点编号从 1 到 n。请编写一个函数，找出并返回所有从节点 1 到节点 n 的路径。每条路径应以节点编号的列表形式表示。

相关描述

【输入描述】

第一行包含两个整数 N，M，表示图中拥有 N 个节点，M 条边

后续 M 行，每行包含两个整数 s 和 t，表示图中的 s 节点与 t 节点中有一条路径

【输出描述】

输出所有的可达路径，路径中所有节点的后面跟一个空格，每条路径独占一行，存在多条路径，路径输出的顺序可任意。

如果不存在任何一条路径，则输出 -1。

注意输出的序列中，最后一个节点后面没有空格！ 例如正确的答案是 1 3 5,而不是 1 3 5， 5后面没有空格！

【输入示例】

5 5

1 3

3 5

1 2

2 4

4 5

【输出示例】

1 3 5

1 2 4 5

• 构造矩阵存储节点

vector<vector<int>> graph(n + 1, vector<int>(n + 1, 0));

while (m--) {

cin >> s >> t; // 使用邻接矩阵 ，1 表示 节点s 指向 节点t

graph[s][t] = 1;

}

• 确认递归函数，参数

vector<vector<int>> result; // 收集符合条件的路径

vector<int> path; // 0节点到终点的路径 // x：目前遍历的节点 // graph：存当前的图 // n：终点

void dfs (const vector<vector<int>>& graph, int x, int n) {

• 确认终止条件

// 当前遍历的节点x 到达节点n

if (x == n) {

// 找到符合条件的一条路径

result.push_back(path);

return;

}

• 处理当前节点出发的所有路径

for (int i = 1; i <= n; i++) { // 遍历节点x链接的所有节点

if (graph[x][i] == 1) { // 找到 x链接的节点

path.push_back(i); // 遍历到的节点加入到路径中来

dfs(graph, i, n); // 进入下一层递归

path.pop_back(); // 回溯，撤销本节点 }

}

• 输出结果

// 输出结果

if (result.size() == 0) cout << -1 << endl;

for (const vector<int> &pa : result) {

for (int i = 0; i < pa.size() - 1; i++) {

// 这里指打印到倒数第二个

cout << pa[i] << " ";

}

cout << pa[pa.size() - 1] << endl; // 这里再打印倒数第一个，控制最后一个元素后面没有空格

}

完整代码：

#include <iostream>
#include <vector>
using namespace std;
vector<vector<int>> result; // 收集符合条件的路径
vector<int> path; // 1节点到终点的路径

void dfs (const vector<vector<int>>& graph, int x, int n) {
    // 当前遍历的节点x 到达节点n 
    if (x == n) { // 找到符合条件的一条路径
        result.push_back(path);
        return;
    }
    for (int i = 1; i <= n; i++) { // 遍历节点x链接的所有节点
        if (graph[x][i] == 1) { // 找到 x链接的节点
            path.push_back(i); // 遍历到的节点加入到路径中来
            dfs(graph, i, n); // 进入下一层递归
            path.pop_back(); // 回溯，撤销本节点
        }
    }
}

int main() {
    int n, m, s, t;
    cin >> n >> m;

    // 节点编号从1到n，所以申请 n+1 这么大的数组
    vector<vector<int>> graph(n + 1, vector<int>(n + 1, 0));

    while (m--) {
        cin >> s >> t;
        // 使用邻接矩阵 表示无线图，1 表示 s 与 t 是相连的
        graph[s][t] = 1;
    }

    path.push_back(1); // 无论什么路径已经是从0节点出发
    dfs(graph, 1, n); // 开始遍历

    // 输出结果
    if (result.size() == 0) cout << -1 << endl;
    for (const vector<int> &pa : result) {
        for (int i = 0; i < pa.size() - 1; i++) {
            cout << pa[i] << " ";
        }
        cout << pa[pa.size() - 1]  << endl;
    }
}