leetcode#124 二叉树中的最大路径和
题目:
路径 被定义为一条从树中任意节点出发,沿父节点-子节点连接,达到任意节点的序列。同一个节点在一条路径序列中 至多出现一次 。该路径 至少包含一个 节点,且不一定经过根节点。
路径和 是路径中各节点值的总和。
给你一个二叉树的根节点 root ,返回其 最大路径和 。
示例:
输入:root = [1,2,3]
输出:6
解释:最优路径是 2 -> 1 -> 3 ,路径和为 2 + 1 + 3 = 6
思路:
1、递归
可知,一个路径总会经过一个中点,也就是拐点,即左子树和右子树的一个LCA。
那么我们就可以把当前节点作为拐点,求左右子树的最值,然后进行计算以当前节点为拐点的答案,最后求最大值即可。
我们定义和值为从根节点到当前节点的路径和。
不难想到,一个 ^
形状的树杈,他的答案是左子树的最大和值和右子树的最大和值减去2倍当前节点的和值加上当前节点的值。
然而题目规定为路径,可以有 /
形状,\
形状,.
形状的树杈,因此,左子树的最大和值和右子树的最大和值,可以与当前节点重合。
步骤:
一、每次到达一个节点时,维护一个节点的和值。
二、然后遍历左子树和右子树,找子树最大和值(包含当前节点)。
三、记录答案ans=左子树的最大和值和右子树的最大和值减去2倍当前节点的和值加上当前节点的值。
四、返回当前节点为根的树的最大和值,也就是左右子树的最大和值和当前节点的和值的最大值。
代码:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution
{
public:
int ans = -1005;
int dfs(TreeNode *now, int sum)
{
if (!now)
return sum;
int sumnow = sum + now->val;
int maxl = max(dfs(now->left, sumnow),sumnow);
int maxr = max(dfs(now->right, sumnow),sumnow);
ans=max(ans,maxl+maxr-2*sumnow+now->val);
return max(maxl, max(maxr, sumnow));
}
int maxPathSum(TreeNode *root)
{
dfs(root, 0);
return ans;
}
};