Problem: 124. 二叉树中的最大路径和
解题方法
递归实现
最大路径和只会出现在以下3种情况:
- 只取当前节点
- 取当前节点和最大的一个孩子
- 取两个孩子,并以当前节点为根节点(这种无需return给上一层)
递归时,每次都记录下上面3种情况的最大值即可
复杂度
时间复杂度:
添加时间复杂度, 示例: O ( n ) O(n) O(n)
空间复杂度:
添加空间复杂度, 示例: O ( 1 ) O(1) O(1)
Code
class Solution {
public:
int res;
int dp(TreeNode* root){// 递归:包含当前节点的路径最大值
if(!root)return -10005;
int leftMax = dp(root->left);
int rightMax = dp(root->right);
int childMax = max(leftMax,rightMax);
// 最大路径和只会出现在以下3种情况
int a = root->val; // 只取当前节点
int b = root->val+childMax; // 取当前节点和最大的一个孩子
int c = root->val+leftMax+rightMax; // 取两个孩子,并以当前节点为根节点(这种无需return给上一层)
// 每次都记录下上面3种情况的最大值
int tmp = max(max(a,b),c);
if(tmp > res)res=tmp;
return max(a,b);// 返回给上一层节点用来做路径的一部分,因为c这种情况已经是一种路径了,就不用返回。
}
int maxPathSum(TreeNode* root) {
res=-10005;
dp(root);
return res;
}
};