给出一棵二叉树,寻找一条路径使其路径和最大,路径可以在任一节点中开始和结束(路径和为两个节点之间所在路径上的节点权值之和)
样例
给出一棵二叉树:
1
/ \
2 3
返回 6
分析:每条最长路径都肯定会以某个顶点为跟,然后两边是以那个节点为跟到叶子节点的最长路径。
代码:
/**
* Definition for binary tree
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
int maxPathSum(TreeNode *root) {
int ret = INT_MIN;
onePath(root,ret);
return ret;
}
int onePath(TreeNode* root,int&ret)
{
if(root==nullptr)
return 0;
int l = onePath(root->left,ret);
int r = onePath(root->right,ret);
ret = max(ret,max(0,l)+max(0,r)+root->val);
return max(0,max(l,r))+root->val;
}
};