题目:
Given a binary tree, find the maximum path sum.
The path may start and end at any node in the tree.
For example:
Given the below binary tree,
1
/ \
2 3
Return 6.
题意:
给定一棵二叉树,找出从任意一个起点,到达任意一个终点,这条路上的值之和最大。
思路:
考虑一个节点node_i,最终最长的那个路径可能是经过 node_i的,也可能是不经过node_i。 对于经过node_i的情况,那么可能包含了node_i的左子树,也有可能包含了node_i的右子树,也有可能两个子树都包括,也可能都不包括。对于不经过node_i,那么结果就是在其左子树中,或者在其右子树中。
我们使用递归的方法,并且通过引用传递一个当前找到的路径的最大值。然后在下一个查找中,去比较与这个最大值的大小关系。
该maxSum函数返回的是通过这个节点,并且路径包含左子树或者右子树(不同时包括两边的子树,如果同时包含了两边的子树,那么该节点的父节点就没法作为最终路径的一部分,因为路径如果同时包含该节点的左子树跟右子树,那么就无法从该节点到达父节点了)。或者所有节点都不包含的路径的最长长度(这个的意思是,通过这个节点无论往左子树或者往右子树延伸,其和都小于0,这个时候就是舍弃该节点及其所有的子树,因为不用这些节点那么这块贡献值是0,但如果用了它们,贡献值反而变成了负的)。
int left = maxSum(root->left, max_result);
int right = maxSum(root->right, max_result);
这两个函数首先去计算该节点的左右子树的最大路径值。而同时会更新max_result值,因为最大的路径可能出现在不包含node_i的部分,也就是说在node_i的左子树或者右子树之中。
max_result = max(max_result, left + right + root->val);
这行代码意思就是,最长路径通过了node_i,可能包含左右子树,也可能不包括。如上面我们介绍的,如果左右子树贡献值为负数,我们就令其贡献0,就是舍弃之。
最后一行作为该函数的返回值, return max(0,max(left,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_result = root->val;
maxSum(root, max_result);
return max_result;
}
int maxSum(TreeNode* root, int& max_result){
if(root == NULL)return 0;
int left = maxSum(root->left, max_result);
int right = maxSum(root->right, max_result);
max_result = max(max_result, left + right + root->val);
return max(0,max(left,right) + root->val);
}
};