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

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

路径 被定义为一条从树中任意节点出发，沿父节点-子节点连接，达到任意节点的序列。
同一个节点在一条路径序列中 至多出现一次 。
该路径 至少包含一个 节点，且不一定经过根节点。
路径和 是路径中各节点值的总和。
给你一个二叉树的根节点 root ，返回其 最大路径和 。
 
示例 1：

输入：root = [1,2,3]
输出：6
解释：最优路径是 2 -> 1 -> 3 ，路径和为 2 + 1 + 3 = 6

示例 2：

输入：root = [-10,9,20,null,null,15,7]
输出：42
解释：最优路径是 15 -> 20 -> 7 ，路径和为 15 + 20 + 7 = 42


提示：
树中节点数目范围是 [1, 3 * 104]
-1000 <= Node.val <= 1000

解答
遍历二叉树
首先获取每个节点中的最大值，这是答案的一种可能，情况1
还有一种和就是左子树和加根节点解右子树的和，情况2
还有一种就是选择左子树和加上根节点或者右子树和加上根节点，选择和更大的组合，情况3

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int ans = -999999;
    int dfs(TreeNode* root){
        if(root==NULL) return 0;

        ans = max(ans,root->val);

        int lsum = dfs(root->left);
        int rsum = dfs(root->right);

        ans = max(ans,max(0,lsum)+root->val+max(0,rsum));

        return max(root->val+max(0,rsum),root->val+max(0,lsum));
    }
    int maxPathSum(TreeNode* root) {
        ans = max(ans,dfs(root));
        return ans;
    }
};