一、题目描述:124. 二叉树中的最大路径和(困难)
给定一个非空二叉树,返回其最大路径和。
本题中,路径被定义为一条从树中任意节点出发,达到任意节点的序列。该路径至少包含一个节点,且不一定经过根节点。
示例 1:
输入: [1,2,3]
1
/ \
2 3输出: 6
示例 2:输入: [-10,9,20,null,null,15,7]
-10
/ \
9 20
/ \
15 7输出: 42
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-maximum-path-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二、解题思路
递归查找左右子树的最大路径和,left,right,由于是寻找某个点到另外一个点的最大值。所以如果left或者right小于0的时候应该抛弃相加小于0的子树。此时应该left = max(left,0),right = max(right,0).然后判断最大值max_sum = max(max_sum , root->val + left,+right)。同时返回max(left+root->val,right+root->val)返回给上游。直到遍历完所有子树,统计所有节点加上左右子树的最大和,就是二叉树的最大路径和。
三、代码
/**
* 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 maxPathSum(TreeNode* root) {
if(root == NULL)
return 0;
int max_sum = INT_MIN;
int root_max = maxPathSum(root,max_sum);
return max_sum;
}
int maxPathSum(TreeNode* root,int & max_sum) {
if(root == NULL)
return 0;
if(root->left == NULL && root->right == NULL)
{
if(root->val > max_sum)
{
max_sum = root->val;
}
return root->val;
}
int left = 0;
int right = 0;
if(root->left != NULL)
{
left = maxPathSum(root->left,max_sum);
}
if(root->right != NULL)
{
right = maxPathSum(root->right,max_sum);
}
if(left < 0)
left = 0;
if(right < 0)
right = 0;
int sum_left_right = left + right + root->val;
//cout<<"root="<<root->val<<endl;
//cout<<"sum_left_right="<<sum_left_right<<endl;
if(sum_left_right > max_sum)
max_sum = sum_left_right;
if(root->left == NULL)
return right + root->val;
if(root->right == NULL)
return left + root->val;
int r_max = max(left,right);
return r_max + root->val;
}
};