二叉树中和为某一值的路径（剑指offer第25题）

最新推荐文章于 2020-04-13 18:56:24 发布

原创最新推荐文章于 2020-04-13 18:56:24 发布 · 200 阅读

0 ·

CC 4.0 BY-SA版权

剑指offer（刷题java）专栏收录该内容

65 篇文章

订阅专栏

本文介绍了一种算法，用于查找二叉树中所有节点值之和等于特定整数的路径。采用前序遍历方式，通过递归实现路径搜索，并在叶子节点处判断路径总和是否与目标值相等。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、题目描述

输入一棵二叉树和整数，打印出二叉树中节点值得和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。

二、解题思路

路径从根节点开始，应该用类似于前序遍历的方式访问树节点。我们需要整个路径，就需要一个容器保存经过路径上的节点，以及一个变量记录当前已有节点元素的和。当前序遍历到某一个节点时，添加该节点到路径，累加节点值。如果该节点为叶子节点并节点值累计等于目标整数，则找到一条路径。如果不是叶子节点，则继续访问子节点。一个节点访问结束后，递归函数自动回到其父节点。
三、java代码

public class Test {

    private ArrayList<Integer> path = new ArrayList<Integer>();

    private ArrayList<ArrayList<Integer>> pathList = new ArrayList<ArrayList<Integer>>();

    public ArrayList<ArrayList<Integer>> findPath(TreeNode root, int sum) {
        if (root == null) {
            return pathList;
        }
        path.add(root.val);
        sum -= root.val;
        if (sum == 0 && root.left == null && root.right == null) {
            pathList.add(new ArrayList<Integer>(path));
        }
        if (root.left != null)
            findPath(root.left, sum);
        if (root.right != null)
            findPath(root.right, sum);
        path.remove(path.size() - 1);
        return pathList;
    }   
}