【Leetcode】124. Binary Tree Maximum Path Sum

题目地址：

https://leetcode.com/problems/binary-tree-maximum-path-sum/

求一棵二叉树中的一条和最大的路径。路径必须至少含一个节点，题目保证树非空。

基本思路是枚举路径最高点。由于要求最大和，我们必须得算出来从某个节点出发向下的所有路径中，和最大是多少，然后我们就可以知道穿过树根的最大和是多少。在计算这个的同时，实时更新当前已经得到的最大和，全局最大和可以用一个全局变量存储即可。代码如下：

class Solution {
 public:
  int maxPathSum(TreeNode *root) {
    int res = INT_MIN;
    dfs(root, res);
    return res;
  }

  int dfs(TreeNode *root, int &res) {
    if (!root) return 0;
    int l = max(0, dfs(root->left, res)), r = max(0, dfs(root->right, res));
    res = max(res, l + r + root->val);
    return root->val + max(l, r);
  }
};

时间复杂度$O(n)$，空间$O(h)$。