力扣-543. 二叉树的直径(Java)(困难)

题目描述:

路径 被定义为一条从树中任意节点出发,沿父节点-子节点连接,达到任意节点的序列。同一个节点在一条路径序列中 至多出现一次 。该路径 至少包含一个 节点,且不一定经过根节点。

路径和 是路径中各节点值的总和。

给你一个二叉树的根节点 root ,返回其 最大路径和 。

示例 1:

输入:root = [1,2,3]
输出:6
解释:最优路径是 2 -> 1 -> 3 ,路径和为 2 + 1 + 3 = 6

示例 2:

输入:root = [-10,9,20,null,null,15,7]
输出:42
解释:最优路径是 15 -> 20 -> 7 ,路径和为 15 + 20 + 7 = 42

提示:

树中节点数目范围是 [1, 3 * 10^4]
-1000 <= Node.val <= 1000

思路:

题目的思路和543. 二叉树的直径的思路一样。
先说543的思路吧:
543题目要求:是给定一棵二叉树,你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。
重点是可能不穿过根节点,所以要遍历所有的节点。
第一个思路就是对每个节点求左子树的深度和右子树的深度之和。
代码如下:

class Solution {
    int sum =0;
    public int diameterOfBinaryTree(TreeNode root) {
        //找到两个最长的节点。
        //可以递归,每个找到自己右子树和左子树最长的。
        if(root == null) return 0;
        if(maxDepth(root.left) + maxDepth(root.right) > sum)
        sum = maxDepth(root.left) + maxDepth(root.right);
        diameterOfBinaryTree(root.left);
        diameterOfBinaryTree(root.right);
        return sum;
    }
    public int maxDepth(TreeNode root) {
        //DFS
        if(root == null) return 0;
        else {
            int left = maxDepth(root.left);
            int right = maxDepth(root.right);
            return Math.max(left, right) + 1;//加1是精髓
        }
    }
}

但是这样用时太长了,所以还是用DFS吧!代码如下:

class Solution {
    int max = 0;
    public int diameterOfBinaryTree(TreeNode root) {
        if (root == null) {
            return 0;
        }
        dfs(root);
        return max;
    }
    int dfs(TreeNode root) {
        if(root == null) return 0;
        //如果root为空,则左子树长度为0,不然继续对左子树深搜,关键是要+1。
        int leftval = root.left == null ? 0 : dfs(root.left) + 1;
        //同左子树
        int rightval = root.right == null ? 0 : dfs(root.right) + 1;
        //更新max
        max = Math.max(max, leftval + rightval);
        //返回左子树和右子树最大的一个边。
        return Math.max(leftval, rightval);
    }  
}

接下来就是本题的思路了,本题和上题是一样的,只不过不用求边长了,而是求最大值,注意子树的最大值可能为负数,如果为负数就不要了。所以要判断一下:int leftval = Math.max(0, dfs(root.left));

class Solution {
    //因为节点可能全为负数,所以要定义max要定为int的最小值。
    int max = Integer.MIN_VALUE;
    public int maxPathSum(TreeNode root) {
        if(root == null) return 0;
        dfs(root);
        return max;
    }
    int dfs(TreeNode root) {
        if(root == null) return 0;
        //如果左子树为负数,那就不走那边了,对应的值咱就不要了。
        int leftval = Math.max(0, dfs(root.left));
        //如果右子树为负数,那就不走那边了,对应的值咱就不要了。
        int rightval = Math.max(0, dfs(root.right));
        //更新最大值,记得加上根节点的值。
        max = Math.max(max, leftval + rightval + root.val);
        //返回根节点的值加上两个子树中值较大的一个。
        return Math.max(leftval, rightval) + root.val;
    }  
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值