124. 二叉树中的最大路径和(leetcode)

给定一个非空二叉树,返回其最大路径和。
本题中,路径被定义为一条从树中任意节点出发,达到任意节点的序列。该路径至少包含一个节点,且不一定经过根节点。
示例 1:
输入: [1,2,3]

       1
      / \
     2   3

输出: 6

示例 2:
输入: [-10,9,20,null,null,15,7]

   -10
   / \
  9  20
    /  \
   15   7

输出: 42

考虑一条路径,可以从任意节点开始,每个节点最多经过一次,问经过的节点的和最大是多少。

思路

递归
首先看到二叉树的题,肯定就是想递归了。递归常规的思路,肯定是递归考虑左子树的最大值,递归考虑右子树的最大值。

public int maxPathSum(TreeNode root) {
    if (root == null) {
        return Integer.MIN_VALUE;
    }
    //左子树的最大值
    int left = maxPathSum(root.left);
    //右子树的最大值
    int right = maxPathSum(root.right);  
    //再考虑包含根节点的最大值
    int  all = ....;
    return Math.max(Math.max(left, right), all);
}

问题就来了,怎么考虑包含根节点的最大路径等于多少?因为我们递归求出来的最大 left 可能不包含根节点的左孩子,例如下边的情况。

     8
    / \
  -3   7
 /  \
1    4

左子树的最大值 left 肯定就是 4 了,然而此时的根节点 8 并不能直接和 4 去相连。所以考虑包含根节点的路径的最大值时,并不能单纯的用 root.val + left + right。
所以如果考虑包含当前根节点的 8 的最大路径,首先必须包含根结点的左右孩子,其次每次遇到一个分叉,就要选择能产生更大的值的路径。例如下边的例子:

      8
    /  \
   -3   7
 /    \
1      4
 \    / \    
  3  2   6

考虑左子树 -3 的路径的时候,我们有左子树 1 和右子树 4 的选择,但我们不能同时选择
如果同时选了,路径就是 ... -> 1 -> -3 -> 4 -> ... 就无法通过根节点 8 了
所以我们只能去求 -3 的左子树能返回的最大值,-3 的右子树能返回的最大值,选一个较大的
假设我们只考虑通过根节点 8 的最大路径是多少,那么代码就可以写出来了。

// 包含根结点的最大路径
public int maxPathSum(TreeNode root) {
    //如果最大值是负数,我们选择不选
    int left = Math.max(helper(root.left), 0);
    int right = Math.max(helper(root.right), 0); 
    return root.val + left + right;
}

int helper(TreeNode root) {
    // 递归出口, 一直计算到叶子结点的左右儿子 
    if (root == null) {
        return 0; 
    }
    int left = Math.max(helper(root.left), 0);
    int right = Math.max(helper(root.right), 0);  
    //选择左子树和右子树产生的值较大的一个
    return root.val + Math.max(left, right);
}

接下来我觉得就是这道题最精彩的地方了,现在我们只考虑了包含最初根节点 8 的路径。那如果不包含当前根节点,而是其他的路径呢?

可以发现在 helper 函数中,我们每次都求了当前给定的节点的左子树和右子树的最大值,和我们 maxPathSum 函数的逻辑是一样的。所以我们利用一个全局变量,在考虑 helper 函数中当前 root 的时候,同时去判断一下包含当前 root 的路径的最大值。

这样在递归过程中就考虑了所有包含当前节点的情况。

代码

int max = Integer.MIN_VALUE;

public int maxPathSum(TreeNode root) {
    helper(root);
    return max;
}

// 找到包含根结点的最大路径和
int helper(TreeNode root) {
    if (root == null) {
        return 0;
    }

    // left 路径的最大值和 0 比较,对于父节点而言,包含 left 结点的最大路径和为负数不如没有
    // 所以这个 left 的定义可以理解为对于父节点有用的最大路径
    int left = Math.max(helper(root.left), 0);
    int right = Math.max(helper(root.right), 0);
    
    // 求的过程中考虑包含当前根节点的最大路径
    max = Math.max(max, root.val + left + right);
    
    // 只返回包含当前根节点和左子树或者右子树的路径
    return root.val + Math.max(left, right);
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 202,056评论 5 474
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 84,842评论 2 378
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 148,938评论 0 335
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,296评论 1 272
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,292评论 5 363
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,413评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,824评论 3 393
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,493评论 0 256
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,686评论 1 295
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,502评论 2 318
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,553评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,281评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,820评论 3 305
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,873评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,109评论 1 258
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,699评论 2 348
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,257评论 2 341

推荐阅读更多精彩内容