给你一棵二叉树,它的根为 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] 之间。
1339. 分裂二叉树的最大乘积 - 力扣(Leetcode)
思路:
第一步计算所有节点之和 totalSum
第二步,遍历所有节点,并计算每个节点的孩子节点下的和 sum, 则乘积= ((totlaSum - sum)%mod) * (sum % mod)
c++
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
int64_t result = 0;
int64_t mod = 1000000007;//((int64_t)pow(10.0, 9) + 7);
int maxProduct(TreeNode* root) {
if(root == nullptr) {
return 0;
}
int64_t totalSum = getNodesSum(root);
dfs(root, totalSum);
return result;
}
void dfs(TreeNode* root, int64_t totalSum) {
if(root == nullptr) {
return;
}
if(root->left != nullptr) {
int64_t sum = getNodesSum(root->left);
result = max(result, ((totalSum - sum) % mod) * (sum % mod ) );
}
if(root->right != nullptr) {
int64_t sum = getNodesSum(root->right);
result = max(result, ((totalSum - sum) % mod) * (sum % mod ) );
}
dfs(root->left, totalSum);
dfs(root->right, totalSum);
}
// 求以 root节点为根节点的所有节点之和
int64_t getNodesSum(TreeNode* root) {
if(root == nullptr) {
return 0;
}
return root->val + getNodesSum(root->left) + getNodesSum(root->right);
}
};