题目如下
这题乍一看,我是没有想到啥好思路的。最后忍不住看了下题解。忍不住赞叹了一句,妙呀!!
解法:其实思路也不难。我们要站在根节点的角度思考问题。我们要从左右节点中获得什么信息,才能得到最终结果。很明显我们最终的路径应该是一条类似倒v形的路径。而倒v的根节点我们开始是不能确定的。这是所有可能结果中取最大值的情况。假设我们现在再某个节点。我们只要知道以左节点为出发向下和以右节点出发向下所经历的路径的最大值。我们就可以得到两个信息,一个是以当前节点出发向下得到的单挑路径的最大值和以当前节点为倒V的根节点的路径的最大值。这个最大最我们可以以全局变量的形式记录。在递归过程中不断更新即可。
代码如下
class Solution {
public:
int maxSum = INT_MIN;
int maxGain(TreeNode *root){
if(root==nullptr){
return 0;
}
int l = max(maxGain(root->left), 0);
int r = max(maxGain(root->right), 0);
int sum = root->val+l+r;
maxSum = max(sum,maxSum);
return root->val + max(l, r);
}
int maxPathSum(TreeNode* root) {
maxGain(root);
return maxSum;
}
};
做二叉树的题目的时候,记住很重要的一点:我们要站在根节点的角度思考问题,需要从左右子节点中获取什么信息。