124. 二叉树中的最大路径和
给定一个非空二叉树,返回其最大路径和。
本题中,路径被定义为一条从树中任意节点出发,达到任意节点的序列。该路径至少包含一个节点,且不一定经过根节点。
示例 1:
输入: [1,2,3]
1
/ \
2 3
输出: 6
示例 2:
输入: [-10,9,20,null,null,15,7]
-10
/ \
9 20
/ \
15 7
输出: 42
解题思路: 此题解题的关键是,要将最大路径分离出两种情况,即以节点出发的最大路径,和穿过这个节点的最大路径,然后前者又可以分路径只包含节点本身和包含节点+若干左(或者右)子树上的节点。另外为了减少遍历过程中的重复计算,因此需要加入记忆数组。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
int helper(TreeNode* node, unordered_map<TreeNode*, int>& m, int &res) {
if (!node) return 0;
if (m.count(node)) {
res = max(res, m[node]);
return m[node];
}
int left = helper(node->left, m, res);
int right = helper(node->right, m, res);
m[node] = max(node->val, max(node->val + left, node->val + right));
res = max(res, max(m[node], left + right + node->val));
return m[node];
}
int maxPathSum(TreeNode* root) {
if (!root) return 0;
unordered_map<TreeNode*, int> m;
int res = INT_MIN;
helper(root, m, res);
return res;
}
};