给定一个非空二叉树,返回其最大路径和。
本题中,路径被定义为一条从树中任意节点出发,达到任意节点的序列。该路径至少包含一个节点,且不一定经过根节点。
示例 1:
输入: [1,2,3] 1 / \ 2 3 输出: 6
示例 2:
输入: [-10,9,20,null,null,15,7] -10 / \ 9 20 / \ 15 7 输出: 42
思路就是dp,每个节点记录该节点以下的最大的权值,若子节点能联通每个节点,则等于子链的权值加上该节点的权值,有一种可能是最大权值就是在某个子链上,不经过上面的主节点,则需要有一个子链最大权值来记录子链的最大权值。
int get_ans(TreeNode* root, int& deep)
{
if (root->left == NULL && root->right == NULL)
{
deep = root->val;
return deep;
}
else if (root->left != NULL && root->right == NULL)
{
int deep1;
int l = get_ans(root->left, deep1);
deep = max(deep1 + root->val, root->val);
return max( max(deep1 + root->val, l), root->val );
}
else if (root->left == NULL && root->right != NULL)
{
int deep2;
int r = get_ans(root->right, deep2);
deep = max(deep2 + root->val, root->val);
return max( max(deep2 + root->val, r), root->val );
}
int deep1, deep2;
int l = get_ans(root->left, deep1);
int r = get_ans(root->right, deep2);
deep = max( max(deep1, deep2) + root->val, root->val );
int ans0 = max(l, r);
int ans1 = deep1 + root->val;
int ans2 = deep2 + root->val;
int ans3 = deep1 + deep2 + root->val;
int ans4 = root->val;
return max( max( max( max(ans0, ans1), ans2), ans3), ans4);
}
class Solution {
public:
int maxPathSum(TreeNode* root) {
int deep;
return get_ans(root, deep);
}
};