124. 二叉树中的最大路径和
给定一个非空二叉树,返回其最大路径和。
本题中,路径被定义为一条从树中任意节点出发,达到任意节点的序列。该路径至少包含一个节点,且不一定经过根节点。
示例 1:
输入: [1,2,3]
1
/ \
2 3
输出: 6
示例 2:
输入: [-10,9,20,null,null,15,7]
-10
/ \
9 20
/ \
15 7
输出: 42
类似于一个后序遍历二叉树的操作。
思路:对于每一个节点来说,只有以他为根节点,才能连接左右两个子树。所以对于每一个节点 T,以 T 作为根节点的最大结果值等于不以它左子树为根节点最大结果值 加 不以它右子树为根节点的最大结果值。每次求得 T为根节点的最大结果值 与 当前最大结果值相比较 ,取最大的值替换当前最大结果值。
C++代码:
class Solution {
public:
int ans;
int maxPathSum(TreeNode* root)
{
ans = root->val;
dfs(root);
return ans;
}
void dfs(TreeNode* root)
{
if(root == NULL) //如果为空节点,返回
return;
dfs(root->left); //往左遍历
dfs(root->right); //往右遍历
int k;int t = root->val; //k保存当前节点的孩子节点(不作为根节点)可以取得的最大值。t为当前节点值。
if(root->left == NULL && root->right ==NULL)//如果为叶子节点,比较该节点作为根节点与当前最大值。
{
ans = max(ans,root->val);
return;
}
if(root->left == NULL) //只有右孩子。且右孩子大于0.则 k = 右孩子的值
{
k = max(root->right->val,0);
root->val += k; //将当前节点值修改为,不以当前节点为根节点能得到的最大值。
}
else if(root->right == NULL) //只有左孩子
{
k = max(root->left->val,0);
root->val += k;
}
else //两个孩子都有
{
k = max(root->right->val,0) + max(root->left->val,0);
root->val += max(max(root->right->val,0) , max(root->left->val,0));
}
ans = max(ans,t + k); //判断 以当前节点为根节点得到的最大值 与 当前最大值 的大小,更新当前最大值
}
};
代码写的不太好,因为修改了原有树的节点值,比较费空间。应该可以用 int dfs(TreeNode* root) 的形式,返回int值,遇到空返回0。在返回值做做文章,应该可以不用修改节点值就能搞定,懒得写了,就这样吧。这样写应该比较容易理解。