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

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

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

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

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

4.总节点数目为n


例如:
给出如下的二叉树, sum=22  

JZ82 二叉树中和为某一值的路径(一)_第1张图片


返回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);
    }

你可能感兴趣的:(剑指,#,算法,算法)