JZ82 二叉树中和为某一值的路径(一)

最新推荐文章于 2023-12-25 13:03:19 发布

自律信仰

最新推荐文章于 2023-12-25 13:03:19 发布

阅读量261

点赞数

分类专栏：剑指 # 算法文章标签：算法

本文链接：https://blog.csdn.net/hza163/article/details/124661367

版权

算法同时被 2 个专栏收录

115 篇文章 0 订阅

订阅专栏

剑指

28 篇文章 0 订阅

订阅专栏

这篇博客探讨了如何利用深度优先搜索（DFS）算法解决二叉树中是否存在节点值之和等于给定sum的路径问题。作者提供了两种解决方案，一种是基础的DFS遍历，另一种则是通过递归优化，降低空间复杂度至O(树的高度)。通过实例和代码展示了如何在有限时间内找到符合条件的路径。

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

给定一个二叉树root和一个值 sum ，判断是否有从根节点到叶子节点的节点值之和等于 sum 的路径。

1.该题路径定义为从树的根结点开始往下一直到叶子结点所经过的结点

2.叶子节点是指没有子节点的节点

3.路径只能从父节点到子节点，不能从子节点到父节点

4.总节点数目为n

例如：
给出如下的二叉树， sum=22

返回true，因为存在一条路径 5\to 4\to 11\to 25→4→11→2的节点值之和为 22

数据范围：

1.树上的节点数满足 0≤n≤10000

2.每个节点的值都满足 ∣val∣≤1000

要求：空间复杂度 O(n)O(n)，时间复杂度 O(n)O(n)

进阶：空间复杂度 O(树的高度)，时间复杂度 O(n)

示例1

输入：

{5,4,8,1,11,#,9,#,#,2,7},22

返回值：

true

示例2

输入：

{1,2},0

返回值：

false

示例3

输入：

{1,2},3

返回值：

true

示例4

输入：

{},0

返回值：

false

简单DFS遍历。注意结点值可能是负的，不能now>sum时剪枝注意空树特判

我的代码:

    boolean flag=false;
    public boolean hasPathSum (TreeNode root, int sum) {
        if(root==null) return false;//特判
        DFS(root,sum,0);
        return flag;
    }

    public void DFS(TreeNode root, int sum,int now){
        if(flag==true||root==null) return;//不能now>sum  可能有负数
        if(root.left==null&&root.right==null){
            if(now+root.val==sum) flag=true;
            return;
        }
        DFS(root.left,sum,now+root.val);
        DFS(root.right,sum,now+root.val);
    }

参考答案解法：简单多了，对递归的理解还不够啊

小难理解

    public boolean hasPathSum (TreeNode root, int sum) {
        if(root==null) return false;//特判
        if(root.left==null&&root.right==null&&sum-root.val==0)
            return true;
        return hasPathSum(root.left,sum-root.val)||hasPathSum(root.right,sum-root.val);
    }