给你一棵二叉树,它的根为 root
。请你删除 1 条边,使二叉树分裂成两棵子树,且它们子树和的乘积尽可能大。
由于答案可能会很大,请你将结果对 10^9 + 7 取模后再返回。
示例 1:
输入:root = [1,2,3,4,5,6] 输出:110 解释:删除红色的边,得到 2 棵子树,和分别为 11 和 10 。它们的乘积是 110 (11*10)
思路:利用两个递归函数。其中第一个递归函数dfs用来计算二叉树值的总和num。第二个递归函数dfs2用来计算最终答案,在递归中,不断计算当前节点作为根节点的子树的值的总和tmp,然后计算mval=tmp*(num-tmp);,再跟新最大值maxavl。
/*** 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:
int mod=1e9+7;
long long num=0;
long long maxval=0;
void dfs(TreeNode *root)
{if(!root){return ;}num+=root->val;dfs(root->left);dfs(root->right);}
int dfs2(TreeNode *root)
{if(!root){return 0;}long long leftval=dfs2(root->left);long long rightval=dfs2(root->right);long long tmp=leftval+rightval+root->val;long long mval=tmp*(num-tmp);maxval=max(mval,maxval);return tmp;
}int maxProduct(TreeNode* root) {dfs(root);dfs2(root);return maxval%mod;}
};