二叉树中和为某一值的路径（三十五）

最新推荐文章于 2022-07-21 14:30:57 发布

原创最新推荐文章于 2022-07-21 14:30:57 发布 · 251 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树 #算法

剑指offer第二版题解专栏收录该内容

37 篇文章

订阅专栏

本文介绍了一种算法，该算法接收一棵二叉树和一个整数作为输入，然后找出所有从根节点到叶子节点的路径，这些路径上的节点值之和等于给定的整数。通过深度优先搜索(DFS)遍历树并动态调整路径和，最终返回所有满足条件的路径列表。

题目

输入一棵二叉树和一个整数，打印出二叉树中结点值的和为输入整数的所有路径。

从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。

样例在这里插入图片描述
算法

class Solution {
    List<List<Integer>> res = new ArrayList();
    List<Integer> path = new ArrayList();
    public List<List<Integer>> findPath(TreeNode root, int sum) {
        dfs(root,sum);
        return res;
    }
    public void dfs(TreeNode root,int sum){
        if(root == null) return ;
        path.add(root.val);
        sum -= root.val;
        if(root.left == null && root.right == null && sum == 0) {
            //引用调用，需要复制path内容，不然存在ans中的为引用
            List <Integer> tmp = new ArrayList <>();
            tmp.addAll(path);
            res.add(tmp);
        }
        dfs(root.left,sum);
        dfs(root.right,sum);
        path.remove(path.size() - 1);
    }
}