原题如下:
给定一个非空二叉树,返回其最大路径和。
本题中,路径被定义为一条从树中任意节点出发,达到任意节点的序列。该路径至少包含一个节点,且不一定经过根节点。
示例 1:
输入: [1,2,3]
1
/ \
2 3
输出: 6
示例 2:
输入: [-10,9,20,null,null,15,7]
-10
/
9 20
/
15 7
输出: 42
思路:
对于任何一个root而言有以下情况:
1.最大路径中包括该root:那么我们需要知道(1) left::从root的左子树第一个结点开始向下延伸可以达到的最大路径(ps:包括左子树第一个结点,但是不一定要延伸到叶子结点)(2)right::root右子树…(同(1)省略!!) 那么包括该结点的最大路径为left+root->val ,left+root->val+right,root->val+right,root->val 四个中的最大一个
2.最大路径中不包括该root点:题中说明了至少是要包括一个结点的,如果是不包括该root点,那么这个最大路径就一定是在左子树,或者右子树内部,这个最大值在我们递归调用到左子树(或者右子树)的时候 已经被算过了,并且被存在了 total_max内;
代码如下:
ps:digui(TreeNode* root, int &total_max)函数返回的是从root开始(包括root)向下延伸可达的最大路径(这个指的是单方向的!!!只是向左子树或者向右子树方向),total_max存的是,至少!!!!包含该结点可达到的最大路径(这个路径是可以多方向的,可以左右都延伸!!!)
int digui(TreeNode* root, int &total_max){
if (root){
int left = 0, right = 0;
if (root->left) left = digui(root->left,total_max) ;
if (root->right) right = digui(root->right,total_max);
//注意left和right都是指从root->left(right) 开始向下可延伸最大路径(单向)
//这里我们不着急把root->val给加上,下面进行求包含root结点,且多方向(指的
//是向左+向右)的最大路径长
total_max = max(total_max, left + right + root->val);
total_max = max(total_max, left + root->val);
total_max = max(total_max, right + root->val);
total_max = max(total_max, root->val);
left += root->val; right += root->val;
//在这里我们才把left和right的值加上root->val,下面我们即将返回包含root的最大
//单向路径
int tmp = max(left, right); tmp = max(root->val, tmp);
return tmp;
}
else return 0;
}
int maxPathSum(TreeNode* root) {
if (root == nullptr)return 0;
int total_max = INT_MIN;
digui(root, total_max);
return total_max;
}