题目
思路
首先把整个树所有节点值之和算出来;
接着计算以每个节点作为根时其子树的节点值 sum 之和;
这样当计算出一个节点及其子树节点值之和 res 后,若以这个节点将树分为两个树,那么乘积就是 res * (sum- res);
每得到一个答案,就与全局的最优值进行比较;
对于取余的问题,可以将res设为long型,最后输出的时候再进行取模操作;
【递归常见类型】
1、这里递归是带返回值的,可以理解为只要带返回值的递归就是说如果你要求解一个大问题,首先你得求解出一个小问题,用小问题的答案去解决大问题;
2、还有一种是传参递归,这种类型的递归一般用于小问题要依赖于大问题的情况;
代码
class Solution {
public:
int mod = 1e9 + 7;
int sum = 0;
long ans = 0;
//计算总的节点和
void DFS(TreeNode* root){
if(root == NULL) return;
sum += root -> val;
DFS(root -> left);
DFS(root -> right);
return;
}
//计算每个子树的节点和
int subDFS(TreeNode* root){
if(root == NULL) return 0;
int left = subDFS(root -> left);
int right = subDFS(root -> right);
int res = left + right + root -> val;
if( (long) res * (sum - res) > ans) ans = (long) res * (sum - res);
return res;
}
int maxProduct(TreeNode* root) {
DFS(root);
subDFS(root);
return ans % mod;
}
};