Leetcode 1339. Maximum Product of Splitted Binary Tree

题目链接：Maximum Product of Splitted Binary Tree

题目大意：给定一个二叉树，切割某条边，会分割成两个不同的二叉树，要求切割出来的两个子树权值和的乘积最大

题目思路：我们可以知道，不管切割哪一条边，一定有一颗子树是某个节点的左右子树之一，那么剩下的权值就是另一个子树的权值，所以我们只需要找到所有节点的左右子树的权值和，然后用总权值和减去该子树的权值和，这就是两个子树的权值和，做乘积，最大的即是我们想要的乘积，算左右子树的和可以用递归做，算出一个子树和就可以放进一个vector，最后再做统一计算。

时间复杂度&&空间复杂度：O（n）&&O（n）

/**
 * 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:
    long long sum = 0;
    long long MOD = 1e9+7;
    long long maxx = 0;
    vector<long long>vec;
    long long countsum(TreeNode* root){
        if(root == NULL) return 0;
        int l_sum = countsum(root->left);
        int r_sum = countsum(root->right);
        if(l_sum != 0) vec.push_back(l_sum);
        if(r_sum != 0) vec.push_back(r_sum);
        return l_sum+r_sum+root->val;
    }
    long long maxProduct(TreeNode* root) {
        sum = countsum(root);
        for(int i = 0;i < vec.size();i++) maxx = max(maxx,(sum-vec[i])*vec[i]);
        return maxx%MOD;
    }
};