路径总和 II

最新推荐文章于 2022-03-15 17:10:39 发布

zjc4j

最新推荐文章于 2022-03-15 17:10:39 发布

阅读量50

点赞数

分类专栏： # 树文章标签： dfs

本文链接：https://blog.csdn.net/qq_29494693/article/details/118445225

版权

树专栏收录该内容

30 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何使用深度优先搜索（DFS）和广度优先搜索（BFS）解决树形结构中寻找路径和目标和的问题。代码示例展示了两种方法的实现，分别使用了栈和队列作为辅助数据结构。这两种算法在遍历树节点时，针对目标和为零的情况，找出所有可能的路径。

摘要由CSDN通过智能技术生成

题目

题源
在这里插入图片描述

代码

深度优先：

class Solution {
    List<List<Integer>> ret = new LinkedList<List<Integer>>();
    Deque<Integer> path = new LinkedList<Integer>();

    public List<List<Integer>> pathSum(TreeNode root, int targetSum) {
        dfs(root, targetSum);
        return ret;
    }
    private void dfs(TreeNode root, int targetSum) {
        if (root == null) return;
        path.offerLast(root.val);
        targetSum -= root.val;
        if (root.left == null && root.right == null && targetSum == 0) {
            ret.add(new LinkedList<Integer>(path));
        }
        dfs(root.left, targetSum);
        dfs(root.right, targetSum);
        path.pollLast();
    }
}

广度优先：

class Solution {
    List<List<Integer>> ret = new LinkedList<List<Integer>>();
    Map<TreeNode, TreeNode> map = new HashMap<TreeNode, TreeNode>();

    public List<List<Integer>> pathSum(TreeNode root, int targetSum) {
        if (root == null) {
            return ret;
        }

        Queue<TreeNode> ququeNode = new LinkedList<TreeNode>();
        Queue<Integer> ququeSum = new LinkedList<Integer>();
        ququeNode.offer(root);
        ququeSum.offer(0);
        
        while (!ququeNode.isEmpty()) {
            TreeNode node = ququeNode.poll();
            int rec = ququeSum.poll() + node.val;

            if (node.left == null && node.right == null) {
                if (rec == targetSum) {
                    getPath(node);
                }
            } else {
                if (node.left != null) {
                    map.put(node.left, node);
                    ququeNode.offer(node.left);
                    ququeSum.offer(rec);
                }
                if (node.right != null) {
                    map.put(node.right, node);
                    ququeNode.offer(node.right);
                    ququeSum.offer(rec);
                }
            }
        }
        return ret;
    }
    private void getPath(TreeNode node) {
        List<Integer> temp = new LinkedList<Integer>();
        
        while (node != null) {
            temp.add(node.val);
            node = map.get(node);
        }
        Collections.reverse(temp);
        // ret.add(new LinkedList<Integer>(temp));
        ret.add(temp);
    }

}