力扣 797 所有可能的路径 C++

题目描述

给一个有 n 个结点的有向无环图，找到所有从 0 到 n-1 的路径并输出（不要求按顺序）

二维数组的第 i 个数组中的单元都表示有向图中 i 号结点所能到达的下一些结点（译者注：有向图是有方向的，即规定了 a→b 你就不能从 b→a ）空就是没有下一个结点了。

输入：graph = [[1,2],[3],[3],[]]
输出：[[0,1,3],[0,2,3]]
解释：有两条路径 0 -> 1 -> 3 和 0 -> 2 -> 3

作者：爱学习的饲养员
链接：https://leetcode-cn.com/leetbook/read/graph/rqabxv/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

解答思路

这里使用的是层序遍历，由于是第一次做图的层序遍历，在本文章中的代码片段做了详细的注释，并与树的层序遍历做了对比，方便对比学习。感谢爱学习的饲养员的讲解！

class Solution {
public:
    vector<vector<int>> allPathsSourceTarget(vector<vector<int>>& graph) {
        vector<vector<int>> result;//存放结果
        if(graph.size() == 0) return result;//图的大小为0，空图，直接返回result
        queue<vector<int>> que;//类似于树的层序遍历，创建一个队列，与书的区别在于，队列里保存的是一个vector
        que.push({0});//将起点作为vector保存在队列里

        while(!que.empty()){
            vector<int> cur = que.front();//类似于树的层序遍历，当前vector是队列里的队首，取出队首
            que.pop();//弹出队首
            int node = cur.back();//复制cur vector中的最后一个元素
            for(auto it : graph[node]){//遍历该元素对应图中的数组
                cur.push_back(it);//添加数组中的元素
                if(it == graph.size() - 1){//如果该元素为终点数字。在这里也就是n-1
                    result.push_back(cur);//直接将该cur vector存进result中
                }else
                    que.push(cur);//否则存进队列中
                cur.pop_back();//弹出元素，进行下一轮添加
            }
        }

        return result;
    }
};