二叉树中的最大路径和

2019-12-21  本文已影响0人  二进制的二哈

题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-maximum-path-sum

给定一个非空二叉树,返回其最大路径和。

本题中,路径被定义为一条从树中任意节点出发,达到任意节点的序列。该路径至少包含一个节点,且不一定经过根节点。

示例 1:

输入: [1,2,3]

       1
      / \
     2   3

输出: 6

示例 2:

输入: [-10,9,20,null,null,15,7]

   -10
   / \
  9  20
    /  \
   15   7

输出: 42

深度优先解法:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public int maxPathSum(TreeNode root) {
        int[] ans = {root.val};
        caculateSum(root,ans);
        return ans[0];
    }

    private void caculateSum(TreeNode root,int[] ans){
        if (root == null)
            return;
        caculateSum(root.left,ans);
        caculateSum(root.right,ans);
        int leftVal = root.left == null?0:root.left.val;
        int rightVal = root.right == null ? 0 : root.right.val;

        int curVal = root.val;
        int leftSum = root.val + leftVal;
        int rightSum = root.val + rightVal;
        int allSum = curVal + leftVal + rightVal;
        //三条路选一条最大的
        int tmpMax = Math.max(curVal, Math.max(leftSum, rightSum));
        if (allSum > tmpMax){
            //只能更新最大值,而不能更新当前节点的值
            ans[0] = Math.max(ans[0],allSum);
        }
        //修改当前的节点值
        root.val = tmpMax;
        ans[0] = Math.max(ans[0],tmpMax);
    }
}
上一篇下一篇

猜你喜欢

热点阅读