LeetCode 1339.分裂二叉树的最大乘积

题目

题目链接

思路

首先把整个树所有节点值之和算出来；
接着计算以每个节点作为根时其子树的节点值 sum 之和；
这样当计算出一个节点及其子树节点值之和 res 后，若以这个节点将树分为两个树，那么乘积就是 res * （sum- res）；
每得到一个答案，就与全局的最优值进行比较；
对于取余的问题，可以将res设为long型，最后输出的时候再进行取模操作；

【递归常见类型】
1、这里递归是带返回值的，可以理解为只要带返回值的递归就是说如果你要求解一个大问题，首先你得求解出一个小问题，用小问题的答案去解决大问题；
2、还有一种是传参递归，这种类型的递归一般用于小问题要依赖于大问题的情况；

代码

class Solution {
public:
    int mod = 1e9 + 7;
    int sum = 0;
    long ans = 0;
    //计算总的节点和
    void DFS(TreeNode* root){
        if(root == NULL) return;
        sum += root -> val;
        DFS(root -> left);
        DFS(root -> right);
        return;
    }
    //计算每个子树的节点和
    int subDFS(TreeNode* root){
        if(root == NULL) return 0;
        int left = subDFS(root -> left);
        int right = subDFS(root -> right);
        int res = left + right + root -> val;
        if( (long) res * (sum - res) > ans) ans = (long) res * (sum - res);
        return res;
    }
    int maxProduct(TreeNode* root) {
        DFS(root);
        subDFS(root);
        return ans % mod;
    }
};