题目描述:
给定一个二叉树,请计算节点值之和最大的路径的节点值之和是多少。
这个路径的开始节点和结束节点可以是二叉树中的任意节点
例如:
给出以下的二叉树,
1↵ / ↵ 2 3
返回的结果为6
思路:
- 这道题乍一看感觉挺简单,但自己整理思路却怎么也整理不好。最后还是没忍住看了答案,才发现自己连题目都看错了…
- 设置一个全局变量maxvalue (在类里面就是私有成员),递归的条件是:以一个节点node为root节点的最大路径等于这个节点的value加左子树的最大路径值加右子树的最大路径值。
- 为了防止最大路径出现负值的情况,左右子树的最大路径需要和0相比较,若比0小则舍去。
代码:
/**
* Definition for binary tree
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
int maxvalue = 0x80000000;
public:
int maxPathSum(TreeNode *root) {
if (root == NULL)
return 0;
dfstree(root);
return maxvalue;
}
int dfstree(TreeNode *node)
{
if (node == NULL)
return 0;
int left = max(0, dfstree(node->left));
int right = max(0, dfstree(node->right));
maxvalue = max(maxvalue, left+right+node->val);
return max(0, max(left, right)+node->val);
}
};
一开始把题目看成 只能从root结点到叶子节点的某一段路径的最大值呢,所以看到maxvalue = max(maxvalue, left+right+node->val);的时候有点懵,心想怎么还要加左右呢,后来才发现题目看错了!
如果没有看错的话应该也是同样的思路,代码改成:
/**
* Definition for binary tree
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
int maxvalue = 0x80000000;
public:
int maxPathSum(TreeNode *root) {
if (root == NULL)
return 0;
dfstree(root);
return maxvalue;
}
int dfstree(TreeNode *node)
{
if (node == NULL)
return 0;
int left = max(0, dfstree(node->left));
int right = max(0, dfstree(node->right));
maxvalue = max(maxvalue, max(left, right)+node->val);
return max(0, max(left, right)+node->val);
}
};
应该就可以了,不过我也没有验证,如果有错误欢迎指正!