算法——图论:路径,回溯

. - 力扣(LeetCode)

给你一个有 n 个节点的 有向无环图(DAG) ,请你找出所有从节点 0 到节点 n-1 的路径并输出(不要求按特定顺序

graph[i] 是一个从节点 i 可以访问的所有节点的列表(即从节点 i 到节点 graph[i][j]存在一条有向边)。

即给定邻接表求0到n-1的所有路径。

自己的代码:dfs,来到当前节点先记录其路径。如果发现当前节点为最后节点则直接返回。否则,对当前节点的每个邻接节点不断dfs。dfs完成后,将当前节点从路径中删除并返回。

注意:此题无需判断节点是否访问过,因为本题是有向图,对于某个节点,指向该节点可能有好几个节点,所以如果该节点只被访问一次无法得到所有路径。

class Solution {
public:
    vector<vector<int>>res;

    vector<vector<int>> allPathsSourceTarget(vector<vector<int>>& graph) {
        vector<int> visited(graph.size(),0),path;
        dfs(graph,0,path,visited);
        return res;
    }

    void dfs(vector<vector<int>>& graph,int cur,vector<int> &path,vector<int>&visited)
    {
        path.push_back(cur);
        if(cur==graph.size()-1)
        {
            res.push_back(path);
            path.pop_back();
            return;
        }
        for(int i=0;i<graph[cur].size();i++)
        {
            //if(!visited[graph[cur][i]])
            dfs(graph,graph[cur][i],path,visited);
        }
        //visited[cur]=1;
        path.pop_back();
    }
};
官方题解:

我们可以使用深度优先搜索的方式求出所有可能的路径。具体地,我们从 0 号点出发,使用栈记录路径上的点。每次我们遍历到点 n−1,就将栈中记录的路径加入到答案中。

特别地,因为本题中的图为有向无环图(DAG\text{DAG}DAG),搜索过程中不会反复遍历同一个点,因此我们无需判断当前点是否遍历过。

class Solution {
public:
    vector<vector<int>> ans;
    vector<int> stk;

    void dfs(vector<vector<int>>& graph, int x, int n) {
        if (x == n) {
            ans.push_back(stk);
            return;
        }
        for (auto& y : graph[x]) {
            stk.push_back(y);
            dfs(graph, y, n);
            stk.pop_back();
        }
    }

    vector<vector<int>> allPathsSourceTarget(vector<vector<int>>& graph) {
        stk.push_back(0);
        dfs(graph, 0, graph.size() - 1);
        return ans;
    }
};

整体思路差不多。

相关推荐
_.Switch16 分钟前
Python机器学习模型的部署与维护:版本管理、监控与更新策略
开发语言·人工智能·python·算法·机器学习
自由的dream1 小时前
0-1背包问题
算法
2401_857297912 小时前
招联金融2025校招内推
java·前端·算法·金融·求职招聘
良月澪二3 小时前
CSP-S 2021 T1廊桥分配
算法·图论
wangyue44 小时前
c# 线性回归和多项式拟合
算法
&梧桐树夏4 小时前
【算法系列-链表】删除链表的倒数第N个结点
数据结构·算法·链表
QuantumStack4 小时前
【C++ 真题】B2037 奇偶数判断
数据结构·c++·算法
今天好像不上班4 小时前
软件验证与确认实验二-单元测试
测试工具·算法
wclass-zhengge5 小时前
数据结构篇(绪论)
java·数据结构·算法
何事驚慌5 小时前
2024/10/5 数据结构打卡
java·数据结构·算法