leetcode原题链接:二叉树中的最大路径和
上一篇:HO49-二叉树的最近公共祖先
下一篇:HOT51-岛屿数量
题目描述
二叉树中的 路径 被定义为一条节点序列,序列中每对相邻节点之间都存在一条边。同一个节点在一条路径序列中 至多出现一次 。该路径 至少包含一个 节点,且不一定经过根节点。路径和 是路径中各节点值的总和。给你一个二叉树的根节点 root
,返回其 最大路径和 。
示例 1:
输入:root = [1,2,3] 输出:6 解释:最优路径是 2 -> 1 -> 3 ,路径和为 2 + 1 + 3 = 6
示例 2:
输入:root = [-10,9,20,null,null,15,7] 输出:42 解释:最优路径是 15 -> 20 -> 7 ,路径和为 15 + 20 + 7 = 42 提示:
- 树中节点数目范围是
[1, 3 * 104]
-1000 <= Node.val <= 1000
解题方法:凡是涉及到二叉树路径相关的最大、最长、最多的题,都可以采用二叉树"三步曲"
1. 先求根节点作为起始的计算值。
2. 利用二叉树遍历的方法获取每一个节点的计算值。
3. 用一个全局变量(函数引用传递)聚合遍历过程中的计算值。
C++代码
/**
* 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 maxPathSum(TreeNode* root) {
int max_sum = INT_MIN;
traverse_max_path_sum(root, max_sum);
return max_sum;
}
void traverse_max_path_sum(TreeNode* root, int& max_sum) {
if (!root) { //空树
return;
}
traverse_max_path_sum(root->left, max_sum); //root左孩子路径和最大贡献值
traverse_max_path_sum(root->right, max_sum);//root右孩子路径和最大贡献值
int left_path_sum = std::max(maxRootPathSum(root->left), 0);
int right_path_sum = std::max(maxRootPathSum(root->right), 0);
max_sum = std::max(max_sum, root->val+ left_path_sum + right_path_sum);
}
int maxRootPathSum(TreeNode* root) { //计算以root节点为起点的,向下延伸(仅向一个方向,因为给父亲节点贡献),自下向上计算,可以避免重复计算
if (!root) {
return 0;
}
int left_cons_value = 0;
if (root->left) {
left_cons_value = std::max(maxRootPathSum(root->left), 0);//先左子树
}
int right_cons_value = 0;
if (root->right) {
right_cons_value = std::max(maxRootPathSum(root->right), 0);//后右子树
}
return root->val + std::max(left_cons_value, right_cons_value);//再计算根节点可贡献的最大值
}
};