1.题目描述
给定一个非空二叉树,返回其最大路径和。
本题中,路径被定义为一条从树中任意节点出发,沿父节点-子节点连接,达到任意节点的序列。该路径至少包含一个节点,且不一定经过根节点。
示例 1:
示例 2:
2.思路
1.路径走入一个子树,能从中获取的最大收益,不关心具体怎么走。这是一种属于递归的、自顶而下的思考方式。
2.返回当前子树能向父节点 “提供” 的最大路径和。即,一条从父节点延伸下来的路径,能在当前子树中获得的最大收益。它分为三种情况,取其中最大的:
(1)停在当前子树的 root,收益:root.val。
(2)走入左子树,最大收益:root.val + dfs(root.left)。
(3)走入右子树,最大收益:root.val + dfs(root.right)。
4.当遍历到 null 节点时,收益为 0,所以返回 0。
5.如果一个子树提供的最大路径和为负,路径走入它,收益不增反减,我希望该子树完全不被考虑,让它返回 0,成为死支。
3.代码
/**
* 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) {
maxGain(root);
return maxSum;
}
int maxGain(TreeNode* root){
if(root == NULL){
return 0;
}
int leftGain = max(maxGain(root->left), 0);
int rightGain = max(maxGain(root->right), 0);
int priceNewPath = root->val + leftGain + rightGain;
maxSum = max(maxSum, priceNewPath);
return root->val + max(leftGain, rightGain);
}
private:
int maxSum = INT_MIN;
};
4.复杂度分析
时间复杂度:O(n)
空间复杂度:O(n)