题目:
给定一个非空二叉树,返回其最大路径和。
本题中,路径被定义为一条从树中任意节点出发,达到任意节点的序列。该路径至少包含一个节点,且不一定经过根节点。
示例 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.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
int max_value = INT_MIN;
int maxPathSum(TreeNode* root) {
max_gain(root);
return max_value;
}
int max_gain(TreeNode* root)
{
if (root == NULL)
return 0;
int left_max = max(max_gain(root->left), 0);
int right_max = max(max_gain(root->right), 0);
int sum = root->val + left_max + right_max;
max_value = max(max_value, sum);
return root->val + max(left_max, right_max);
}
};