LeetCode 124. 二叉树中的最大路径和 递归 + 深度优先搜索 (DFS)详解

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

题目来源

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

题目分析

给定一个非空的二叉树，要求返回其最大路径和。本题中的路径被定义为一条从树中任意节点出发，达到任意节点的序列，且至少包含一个节点，并且路径不一定经过根节点。

题目难度

• 难度：困难

题目标签

• 标签：动态规划, 树

题目限制

• 树中节点的数目范围是 [1, 3 * 10^4]
• -1000 <= Node.val <= 1000

解题思路

思路：递归 + 深度优先搜索 (DFS)

1. 定义问题：

   • 最大路径和的路径可以经过根节点，也可以只在左子树或右子树中。因此，对于每个节点，我们都需要计算以当前节点为根节点的最大路径和。

2. 状态转移：

   • 采用 DFS 的递归方式，计算每个节点为根节点的最大路径和，同时更新全局最大路径和。

3. 递归公式：

   • 对于每个节点，其最大路径和为：max(左子树的最大路径和, 右子树的最大路径和) + 当前节点值。
   • 如果路径包括当前节点的左右子树，则需要计算：left + right + root.val，并更新全局最大路径和 ans。

4. 最终结果：

   • 通过 DFS 遍历整棵树，并在每次递归时更新 ans，最终 ans 即为所求的最大路径和。

核心算法步骤

1. DFS 递归：

   • 从根节点开始，通过 DFS 遍历整棵树，在遍历过程中计算每个节点的左子树和右子树的最大路径和，并更新全局最大路径和 ans。

2. 返回最大路径和：

   • 每次递归返回当前节点为根节点的最大路径和。

代码实现

以下是求解二叉树最大路径和的 Java 代码：

/**
 * 124. 二叉树中的最大路径和
 * @param root 二叉树
 * @return 最大路径和
 */
public int maxPathSum(TreeNode<Integer> root) {
    maxPathSum2(root);
    return ans;
}

private int ans = Integer.MIN_VALUE;

public int maxPathSum2(TreeNode<Integer> root) {
    if (root == null) {
        return 0;
    }
    int left = Math.max(0, maxPathSum2(root.left));
    int right = Math.max(0, maxPathSum2(root.right));
    ans = Math.max(ans, left + right + root.val);
    return Math.max(left, right) + root.val;
}

代码解读

• maxPathSum2(root) 是一个递归函数，用于计算以当前节点为根的子树的最大路径和。
• left 和 right 分别表示当前节点左子树和右子树的最大路径和，如果某一侧路径和为负数，则取0。
• ans 用于存储二叉树的最大路径和，在每次递归过程中更新最大值。

性能分析

• 时间复杂度：O(N)，其中 N 是树的节点数量。每个节点遍历一次。
• 空间复杂度：O(H)，其中 H 是树的高度。递归栈的空间开销。

测试用例

以下是一些测试用例，用于验证代码的正确性：

TreeNode<Integer> root1 = new TreeNode<>(1);
root1.left = new TreeNode<>(2);
root1.right = new TreeNode<>(3);
System.out.println(maxPathSum(root1)); // 输出: 6

TreeNode<Integer> root2 = new TreeNode<>(-10);
root2.left = new TreeNode<>(9);
root2.right = new TreeNode<>(20);
root2.right.left = new TreeNode<>(15);
root2.right.right = new TreeNode<>(7);
System.out.println(maxPathSum(root2)); // 输出: 42

TreeNode<Integer> root3 = new TreeNode<>(2);
root3.left = new TreeNode<>(-1);
System.out.println(maxPathSum(root3)); // 输出: 2

总结

通过递归和 DFS 算法，我们可以有效地求解二叉树中的最大路径和问题。此方法思路清晰，易于实现，且具有较高的效率。

---