题目
给定一个非空二叉树,返回其最大路径和。
本题中,路径被定义为一条从树中任意节点出发,达到任意节点的序列。该路径至少包含一个节点,且不一定经过根节点。
示例 1:
输入: [1,2,3]
1
/ \
2 3
输出: 6
示例2
输入: [-10,9,20,null,null,15,7]
-10
/
9 20
/
15 7
输出: 42
思路
个人感觉本题题意不是很明确(可能是我理解的问题),在题意的思考上花了较多时间。最后还是没有思路,看了官方的答案。。。很多人说这题并不难,不符合hard的难度,可能是我想复杂了吧。这里记录下官方思路。有以下设定:
- 叶子节点的收益等于他的节点值(如果大于0)
- 如果一个节点非叶子节点,则他的收益等于自身的值加上子树中收益较大的一支
- 以一个节点为根的路径,最大路径和为节点值加上左右子树收益(如果大于0)
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
int maxSum = Integer.MIN_VALUE;
public int maxPathSum(TreeNode root) {
maxGain(root);
return maxSum;
}
public int maxGain(TreeNode root) {
if (root == null) {
return 0;
}
int leftGain = Math.max(maxGain(root.left),0);
int rightGain = Math.max(maxGain(root.right),0);
int current = root.val + leftGain + rightGain;
maxSum = Math.max(maxSum, current);
return root.val + Math.max(rightGain , leftGain);
}
}