This approach uses a recursive depth-first search (DFS) to explore every path in the binary tree. We calculate the maximum path sum that a node can contribute to its parent, which is the largest possible sum of the path from this node to any leaf node passing through this node. We keep updating the global maximum path sum during the recursion.
Time Complexity: O(n), where n is the number of nodes in the binary tree. We traverse each node exactly once.
Space Complexity: O(h), where h is the height of the tree. This space is used by the call stack for recursion.
1class TreeNode:
2 def __init__(self, val=0, left=None, right=None):
3 self.val = val
4 self.left = left
5 self.right = right
6
7def maxPathSum(root):
8 def dfs(node):
9 if not node:
10 return 0
11 left_max = max(dfs(node.left), 0)
12 right_max = max(dfs(node.right), 0)
13 current_path_sum = node.val + left_max + right_max
14 dfs.max_sum = max(dfs.max_sum, current_path_sum)
15 return node.val + max(left_max, right_max)
16
17 dfs.max_sum = float('-inf')
18 dfs(root)
19 return dfs.max_sum
We define a nested function dfs
that takes a node as an input and returns the maximum sum of the path that includes the node as an endpoint. The function recursively calculates the maximum path sum of the left and right subtrees. It then calculates the maximum path sum that includes the current node. We maintain a global variable dfs.max_sum
to keep track of the maximum path sum encountered so far. Finally, the function returns the maximum path sum starting at the current node. We initialize the maximum path sum to negative infinity to ensure any path replaces it.
In this approach, we utilize dynamic programming principles alongside tree traversal to compute the maximum path sum. This method is more about reformulating the tree's recursive structure into a form that allows storing intermediate results to optimize operations across the different subtrees. This method is especially useful for understanding different formulations but often overlaps logically with recursive DFS.
Time Complexity: O(n), covering a single pass over all nodes.
Space Complexity: O(h), necessitated by call stack usage designated by the tree height h.
1class TreeNode {
2 int val;
3 TreeNode left;
4 TreeNode right;
5 TreeNode(int x) { val = x; }
6}
7
8public class Solution {
9 int maxSum = Integer.MIN_VALUE;
10 public int maxPathSum(TreeNode root) {
11 calculateMaxSum(root);
12 return maxSum;
13 }
14 private int calculateMaxSum(TreeNode node) {
15 if (node == null) return 0;
16 int leftGain = Math.max(calculateMaxSum(node.left), 0);
17 int rightGain = Math.max(calculateMaxSum(node.right), 0);
18 int priceNewPath = node.val + leftGain + rightGain;
19 maxSum = Math.max(maxSum, priceNewPath);
20 return node.val + Math.max(leftGain, rightGain);
21 }
22}
In this solution, we employ a recursive method calculateMaxSum
that computes sub-results by leveraging already seen sub-tree sub-results. By traversing each node once, we update a global max value maxSum
with results derived from each node acting as a center of a new path. This method hence avoids recomputation by relying on previously calculated results, drawing dynamic programming parallels.