LeetCode解题笔记 30 —— 124. 二叉树中的最大路径和

题目

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

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

示例 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[] max = new int[]{Integer.MIN_VALUE};
        check(root,max, new int[]{Integer.MIN_VALUE});
        return max[0];
    }
    
    private void check(TreeNode root, int[] max, int[] temp){
        if(root==null){
            temp[0] = 0;
            return;
        }
        //左子树到本节点的最大路径
        int[] maxL = new int[]{Integer.MIN_VALUE};
        check(root.left,max, maxL);
        //右子树到本节点的最大路径
        int[] maxR = new int[]{Integer.MIN_VALUE};
        check(root.right,max, maxR);
        int sum = root.val;
        sum =  maxL[0]+root.val > sum ? maxL[0]+root.val : sum;
        sum =  maxR[0]+root.val > sum ? maxR[0]+root.val : sum;
        //本节点的父节点的最大路径
        temp[0] = sum;
        //当前节点最大路径
        sum =  maxL[0]+maxR[0]+root.val > sum ?  maxL[0]+maxR[0]+root.val : sum;
        if(sum > max[0]){
            max[0] = sum;
        }
    }
}