您的位置:首页 > 其它

[LeetCode] Binary Tree Maximum Path Sum

2013-11-08 00:18 495 查看
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
.
这道题其实跟求二叉树的深度类似,只不过需要求是带权值的深度而已,即所说的path sum。对于任意一个节点,其对应的path sum应该等于:其左子树节点的带权值深度 + 其右子树的带权值深度 + 其自身的值。而带权值的深度求法和一般的深度求法类似,就是取子节点带权值的深度的那个,加上自身的值即可。

这里需要注意的是,由于题目中没有说明所有节点的值都为非负数,所以得考虑负数的存在。如果一旦深度为负值,那么计算path sum的时候还不如不加子节点的带权值深度,所以这里每次计算出的带权值的深度,我都会跟0比较,如果比0小,那么我就取0。

// given a nonempty node, retrieve the weighted depth and update the max path sum
int weightedDepth(TreeNode *root, int & maxPathSum) {
int leftPathSum = 0, rightPathSum = 0;
if (root->left) leftPathSum = max(0, weightedDepth(root->left, maxPathSum));
if (root->right) rightPathSum = max(0, weightedDepth(root->right, maxPathSum));
// update the max path sum
maxPathSum = max(maxPathSum, leftPathSum + root->val + rightPathSum);
return max(leftPathSum + root->val, rightPathSum + root->val);
}

int maxPathSum(TreeNode *root) {
if (!root) return 0;
int maxPathSum = INT_MIN;
weightedDepth(root, maxPathSum);
return maxPathSum;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息