题目
给定一个非空二叉树,返回其最大路径和。
本题中,路径被定义为一条从树中任意节点出发,达到任意节点的序列。该路径至少包含一个节点,且不一定经过根节点。
示例 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;
}
}
}