LeetCode——797. 所有可能的路径(Java)

题目描述

题干:
给你一个有 n 个节点的 有向无环图(DAG)
请你找出所有从节点 0 到节点 n-1 的路径并输出(不要求按特定顺序)
二维数组的第 i 个数组中的单元都表示有向图中 i 号节点所能到达的下一些节点
空就是没有下一个结点了,有向图是有方向的,即规定了 a→b 你就不能从 b→a 。

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

示例2:
输入:graph = [[4,3,1],[3,2,4],[3],[4],[]]
输出:[[0,4],[0,3,4],[0,1,3,4],[0,1,2,3,4],[0,1,4]]

示例 3:
输入:graph = [[1],[]]
输出:[[0,1]]

示例 4:
输入:graph = [[1,2,3],[2],[3],[]]
输出:[[0,1,2,3],[0,2,3],[0,3]]

示例 5:
输入:graph = [[1,3],[2],[3],[]]
输出:[[0,1,2,3],[0,3]]

题解思路

返回第一个到最后一个节点的所有路径,而且这里的模型是有向无环图

所以无论用树还是图都可以,这里采用深度优先遍历的思路,当然广度也可以

用队列保存所有可以到达的路径,最后返回总的List集合

正确代码

class allPathsSourceTargetSolution {
    List<List<Integer>> ans = new ArrayList<>();
    Deque<Integer> stack = new ArrayDeque<>();

    public List<List<Integer>> allPathsSourceTarget(int[][] graph) {
        stack.offerLast(0);
        dfs(graph, 0, graph.length - 1);
        return ans;
    }

    public void dfs(int[][] graph, int x, int n) {
        if (x == n) {
            ans.add(new ArrayList<Integer>(stack));
            return;
        }
        for (int y : graph[x]) {
            stack.offerLast(y);
            dfs(graph, y, n);
            stack.pollLast();
        }
    }
}

总结

这里需要总结一下Queue的一些常用方法,感兴趣大家可以去看我的另一篇总结常用方法的文章

https://www.cnblogs.com/bc-song/p/15117117.html

如果文章存在问题或者有更好的题解,欢迎在评论区斧正和评论,各自努力,你我最高处见

posted @ 2021-08-25 09:08  21岁还不是架构师  阅读(109)  评论(0编辑  收藏  举报