LeetCode 124. 二叉树中的最大路径和

作者:Choi Yang

更新于:7 个月前

字数统计:362 字

阅读时长:1 分钟

题目描述

给定一个非空二叉树，返回其最大路径和。

本题中，路径被定义为一条从树中任意节点出发，沿父节点-子节点连接，达到任意节点的序列。该路径至少包含一个节点，且不一定经过根节点。

示例 1：

输入：[1,2,3]

       1
      / \
     2   3

输出：6

示例 2：

输入：[-10,9,20,null,null,15,7]

   -10
   / \
  9  20
    /  \
   15   7

输出：42

来源：力扣（LeetCode） 链接：https://leetcode-cn.com/problems/binary-tree-maximum-path-sum 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

后序遍历，先遍历左孩子，对于孩子的累计和，我们判断一下，如果小于 0（即为负数）就没必要加了，直接返回 0 即可，否则加上孩子累计和。然后我们对每一层求一下最大值即可。

/**
 * Definition for a binary tree node.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number}
 */
var maxPathSum = function (root) {
  let res = Number.MIN_SAFE_INTEGER;
  let dfs = (root) => {
    if (!root) return 0;
    // 后序遍历，先遍历左孩子
    let left = root.left && dfs(root.left);
    let right = root.right && dfs(root.right);
    // 每一层求一下最大值
    res = Math.max(res, left + right + root.val);
    let sum = Math.max(left, right) + root.val;
    // 判断一下如果孩子的累计和小于0，就没必要加了
    return sum < 0 ? 0 : sum;
  };
  dfs(root);
  return res;
};

学如逆水行舟，不进则退

Contributors

Choi Yang