Leetcode 1339.分裂二叉树的最大乘积（Maximum Product of Splitted Binary Tree）

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

1 题目描述（Leetcode题目链接）

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

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

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

提示：

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

2 题解

  我们先求出每个子树的节点和，乘积=当前节点的子树和*总和减去当前节点的子树和。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def sum_of_tree(self, root):
        if not root:
            return 0
        l = self.sum_of_tree(root.left)
        r = self.sum_of_tree(root.right)
        self.seq.append(root.val + l + r)
        return self.seq[-1]

    def maxProduct(self, root: TreeNode) -> int:
        self.seq = []
        self.sum_of_tree(root)
        res = 0
        for num in self.seq:
            res = max((self.seq[-1]-num)*num, res)
        return res%(10**9+7)