leetcode [1339] 分裂二叉树的最大乘积

---

Contact me:

Blog -> https://cugtyt.github.io/blog/index
Email -> cugtyt@qq.com
GitHub -> Cugtyt@GitHub

---

给你一棵二叉树，它的根为 root 。请你删除 1 条边，使二叉树分裂成两棵子树，且它们子树和的乘积尽可能大。

由于答案可能会很大，请你将结果对 10^9 + 7 取模后再返回。

示例 1：

输入：root = [1,2,3,4,5,6]
输出：110
解释：删除红色的边，得到 2 棵子树，和分别为 11 和 10 。它们的乘积是 110 （11*10）

示例 2：

输入：root = [1,null,2,3,4,null,null,5,6]
输出：90
解释：移除红色的边，得到 2 棵子树，和分别是 15 和 6 。它们的乘积为 90 （15*6）

示例 3：

输入：root = [2,3,9,10,7,8,6,5,4,11,1]
输出：1025

示例 4：

输入：root = [1,1]
输出：1

提示：

每棵树最多有 50000 个节点，且至少有 2 个节点。
每个节点的值在 [1, 10000] 之间。

核心思想，把每个子树的和保存下来，这样候选值为(总和-子树和)*子树和。

class Solution:
    def maxProduct(self, root: TreeNode) -> int:
        ans = 0
        mod = 10**9 + 7
        buff = []
        def core(root):
            nonlocal mod
            if not root:
                return 0
            left = core(root.left)
            right = core(root.right)
            buff.append(left + right + root.val)
            return buff[-1]
        core(root)
        
        sum_val = buff[-1]
        for b in buff:
            ans = max(ans, (sum_val - b) * b)
        return ans % mod