主页有其他数据结构内容(持续更新中)
难度:Hard
代码:
/**
* 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 {
private:
int res = INT_MIN; // 由于节点值可以为负数,所以res不能设置为0,要设置为INT_MIN
int maxPath(TreeNode* root) { // 计算以当前节点root为根节点的最大路径
if (root == nullptr) {
return 0;
}
// 分别计算以左、右孩子为根节点的最大路径
// 如果最大路径和<0,意味着该路径和对总路径和做负贡献,因此不要计入到总路径中,将它设置为0
int left = max(maxPath(root->left), 0);
int right = max(maxPath(root->right), 0);
// 比较当前最大路径和与左右子树最长路径加上根节点值的较大值,更新全局变量
res = max(res, left + right + root->val);
// 返回左右子树较长的路径加上根节点值
return max(left + root->val, right + root->val);
}
public:
int maxPathSum(TreeNode* root) {
maxPath(root);
return res;
}
};