给定一个非空二叉树,返回其最大路径和。
本题中,路径被定义为一条从树中任意节点出发,沿父节点-子节点连接,达到任意节点的序列。该路径至少包含一个节点,且不一定经过根节点。
示例 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() : 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 = -1e9;
int dfs(TreeNode *root) {
if(root == NULL) return 0;
int lef = max(dfs(root -> left), 0);//找到左子树中最大路径和
int rig = max(dfs(root -> right), 0);//找到右子树中最大路径和
int val = root -> val + lef + rig;
ans = max(ans, val);//不断更新答案
return root -> val + max(lef, rig);//返回以当前节点为根节点的最大路径和
}
int maxPathSum(TreeNode* root) {
dfs(root);
return ans;
}
};