【刷爆LeetCode】七月算法集训（18）树

最新推荐文章于 2024-10-03 21:46:55 发布

...1999

最新推荐文章于 2024-10-03 21:46:55 发布

阅读量151

点赞数

分类专栏：七月算法集训文章标签： leetcode 算法深度优先

本文链接：https://blog.csdn.net/LAdestiny/article/details/125840493

版权

七月算法集训专栏收录该内容

31 篇文章 0 订阅

订阅专栏

题目来源于知识星球—英雄算法联盟，七月算法集训专题

前言

跟随英雄算法联盟博主—英雄哪里出来，每天完成相应的算法练习，一个月后，必定会有所成长！

一、1339.分裂二叉树的最大乘积（中等）

1.题目描述

2.解题思路

首先计算每个节点当做根节点时，其子树所有节点的累加和，存储在哈希表sum中；然后递归，计算
删除每条边之后的累加和，计算分裂后两个子树的乘积，如果删除a->b，那么乘积为sum[b]*(n-sum[b])。
n为所有节点的累加和。最后对乘积取最大值之后再对 10^9+7取模。

3.代码演示（C++）

class Solution 
{
    unordered_map<TreeNode*,int>sum;
    int n;
    long long ans;
    int dfs(TreeNode *root)
    {
        if(root==NULL)
        {
            return 0;
        }
        int s=root->val;
        return sum[root]=s+dfs(root->left)+dfs(root->right);
    }
    void muldfs(TreeNode *root)
    {
        if(root->left)
        {
            ans=max(ans,(long long)sum[root->left]*(n-sum[root->left]));
            muldfs(root->left);
        }
        if(root->right)
        {
            ans=max(ans,(long long)sum[root->right]*(n-sum[root->right]));
            muldfs(root->right);
        }
    }
public:
    int maxProduct(TreeNode* root) 
    {
        long long num=1e9+7;
         dfs(root);
         n=sum[root];
         ans=0;
         muldfs(root);
         return ans%num;
    }
};