难度:困难。
标签:二叉树,深度优先搜索,动态规划。
findMax返回值是cur节点所在的两条路径(左子树+cur 或 右子树+cur)的最大路径和,因为cur需要向上继续蔓延,所以不包括左子树+cur+右子树这条路径。
在findMax函数中[line 19 - 28],若左右子树的两条路径的最大值大于0,则加上左右子树的路径,此处考虑了cur所在的路径(不往父节点蔓延的路径)的节点最大值。
正确解法:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
int ans;
int findMax(TreeNode* cur){
if(cur == nullptr)return 0;
int leftMax = findMax(cur->left);
int rightMax = findMax(cur->right);
int sum = cur->val;
ans = max(ans, sum);
if(leftMax > 0){
sum += leftMax;
ans = max(ans, sum);
}
if(rightMax > 0){
sum += rightMax;
ans = max(ans, sum);
}
return max(cur->val, max(cur->val + leftMax, cur->val + rightMax));
}
public:
int maxPathSum(TreeNode* root) {
ans = root->val;
findMax(root);
return ans;
}
};
结果: