LeeCode 124 DFS

题意

传送门 LeeCode 124. 二叉树中的最大路径和

题解

对于二叉树的最大路径，$DFS$ 深度相同的节点最多只有 $2$ 个，且 $DFS$ 深度最小的节点只有 $1$ 个。

对于每一个节点，最大路径和可能出现在包含此节点且此节点 $DFS$ 深度最小的路径，或包含此节点但此节点 $DFS$ 深度并非最小的路径。那么对于第一种情况，用节点的值加上左右子树的没有相同 $DFS$ 深度节点的最大路径和，以更新答案；对于第二种情况，将左右子树中，没有相同 $DFS$ 深度的节点的最大路径和（且大于 $0$）加上节点的值回溯到当前节点的父节点。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int res = INT_MIN;
    int dfs(TreeNode *root){
        int chl = root->left != nullptr ? dfs(root->left):0;
        int chr = root->right != nullptr ? dfs(root->right):0;
        res = max(res, root->val + max(0, chl) + max(0, chr));
        return max(0, max(chl, chr)) + root->val;
    }
    int maxPathSum(TreeNode *root) {
        dfs(root);
        return res;
    }
};