给你一棵二叉树,它的根为
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]
之间。
/**
* Definition for a binary tree node.
* function TreeNode(val, left, right) {
* this.val = (val===undefined ? 0 : val)
* this.left = (left===undefined ? null : left)
* this.right = (right===undefined ? null : right)
* }
*/
/**
* @param {TreeNode} root
* @return {number}
*/
let avg,ans = 0;
var maxProduct = function(root) {
let total = getTotal(root);
avg = total / 2;
ans = total;
getTotal(root);
return ans *(total - ans) % (10 ** 9 + 7);
};
var getTotal = function(root) {
if(root == null) return 0;
let val = root.val + getTotal(root.left) + getTotal(root.right);
if(Math.abs(val - avg) < Math.abs(ans - avg)) ans = val;
return val;
};