《七月集训》第十八日——二叉树

最新推荐文章于 2024-07-25 11:27:31 发布

野生的小小风256

最新推荐文章于 2024-07-25 11:27:31 发布

阅读量175

点赞数 1

文章标签：深度优先算法

本文链接：https://blog.csdn.net/smallwind256/article/details/125983890

版权

前言

这是七月集训的第十八日，今日的训练内容是 二叉树

解题报告

今日的题目在写完之后对DFS和递归的理解更深了。

1.力扣1339

原题链接

1339. 分裂二叉树的最大乘积

题目概述

给你一棵二叉树，它的根为 root 。请你删除 1 条边，使二叉树分裂成两棵子树，且它们子树和的乘积尽可能大。

由于答案可能会很大，请你将结果对 10^9 + 7 取模后再返回。

解题思路

首先要先统计出这棵树所有的结点的结点的数据和，最终所需要两个子树的乘积和其实就是将数拆为两个部分然后相乘，如果想要这个数据更大，那么通过基本不等式我们可知，当两数之和确定时，当两数相等时它们的乘积最大，所以我们只要找到与中间的数最接近的数即可。

用于求解当前结点的子树的和的函数使用DFS来实现，在实现了此函数之后，当我们寻找与中间数最接近的时候也可以使用已经实现了的DFS来实现。

源码剖析

int DFS(struct TreeNode* p){
    if(p==NULL){
        return 0;
    }
    int left = DFS(p->left);
    int right = DFS(p->right);
    p->val += left + right; 
    return p->val;
}

void DFS2(struct TreeNode* p,long long half,int *min,long long* ans){
    if(p==NULL){
        return;
    }
    if((abs(p->val-half)<(*min))){
        (*ans) = p->val;
        (*min) = abs(p->val-half);
    }
    DFS2(p->left,half,min,ans);
    DFS2(p->right,half,min,ans);
}

int maxProduct(struct TreeNode* root){
    long long sum = DFS(root);
    int min = INT_MAX;
    long long ans = 0;
    DFS2(root,(sum)/2,&min,&ans);
    printf("%d,%d",sum,ans);
    return (ans*(sum-ans))%1000000007;
}