深度优先搜索——DAG图的所有路径

原题在这里

概述就不用说了,

直接分析:

dfs(x,y):
当前深搜遍历节点为x,使用的ans存储下标为y
1.当x为最后元素时候结束搜索
2.当搜索不是第一条路径时候新建存储路径,所以遍历前存储now路径
3.遍历完后,删除未到达最后元素的路径

因为不熟悉,还是被卡住了

code:

class Solution
{
    vector<vector<int>> mp, ans;
    int l, num;
    void dfs(int x, int y)
    {
        ans[y].emplace_back(x);
        if (x == l - 1)
            return;
        vector<int> now = ans[y];
        for (int i = 0; i < mp[x].size(); ++i)
        {
            if (i != 0)
            {
                ans.emplace_back(now);
                dfs(mp[x][i], ++num);
            }
            else
                dfs(mp[x][i], y);
        }
    }

public:
    vector<vector<int>> allPathsSourceTarget(vector<vector<int>> &graph)
    {
        l = graph.size(), num = 0;
        mp = graph;
        ans.emplace_back(vector<int>());
        dfs(0, 0);
        for (int i = 0; i < ans.size(); ++i)
            if (ans[i][ans[i].size() - 1] != l - 1)
                ans.erase(ans.begin() + i--);
        return ans;
    }
};

 

posted @ 2022-03-22 21:50  Renhr  阅读(379)  评论(0)    收藏  举报