【leetcode】Binary Tree Maximum Path Sum (medium)

#include <iostream>
#include <vector>
#include <algorithm>
#include <queue>
#include <stack>
using namespace std;

//Definition for binary tree
struct TreeNode {
     int val;
     TreeNode *left;
     TreeNode *right;
     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
class Solution {
public:
    int maxPathSum(TreeNode *root){
        if(root == NULL)
        {
            return 0;
        }
        int MaxPathSum = root->val; //赋的初值一定要小于等于最后的值
        maxPathSumCur(root, MaxPathSum);
        return MaxPathSum;
    }
    int maxPathSumCur(TreeNode *root, int& MaxPathSum) {
        if(root == NULL)
        {
            return 0;
        }

        int lsum =  maxPathSumCur(root->left, MaxPathSum);
        int rsum = maxPathSumCur(root->right, MaxPathSum);
        int maxPathSumCurrent = root->val; //每次根的值一定要加上 左右子树的就加大于0的
        if(lsum > 0)
        {
            maxPathSumCurrent += lsum;
        }
        if(rsum > 0)
        {
            maxPathSumCurrent += rsum;
        }
        
        MaxPathSum = max(maxPathSumCurrent, MaxPathSum);
        return max(root->val, max(root->val + lsum, root->val +rsum)); //返回时返回根 节点加左 或右子树 或单独根节点中最大的
    }
    void create(TreeNode *& root)
    {
        int d;
        scanf("%d", &d);
        if(d != 0)
        {
            root = new TreeNode(d);
            create(root->left);
            create(root->right);
        }
    }
};

int main()
{
    Solution s;
    TreeNode * T = NULL;
    s.create(T);
    int sum = s.maxPathSum(T);

    return 0;
}