【二叉树】分裂二叉树的最大乘积

最新推荐文章于 2023-12-25 18:00:00 发布

豪冷啊

最新推荐文章于 2023-12-25 18:00:00 发布

阅读量312

点赞数 1

分类专栏：算法文章标签： swift 二叉树算法

本文链接：https://blog.csdn.net/xjh093/article/details/125888521

版权

算法专栏收录该内容

121 篇文章 0 订阅

订阅专栏

0x00 题目

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

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

0x01 思路

因为要把树分成 2 部分
假设分割后第一棵树的和为 S
则剩下部分的和为整体的和 减去 S
所以需要先求出整棵树的和 T
因为不知道第一棵树以哪个节点为根合适
所以任何节点都需要遍历到
使得 S * (T - S) 最大
同时记录最大值

0x02 解法

语言：Swift

树节点：TreeNode

public class TreeNode {
    public var val: Int
    public var left: TreeNode?
    public var right: TreeNode?
    public init() { self.val = 0; self.left = nil; self.right = nil; }
    public init(_ val: Int) { self.val = val; self.left = nil; self.right = nil; }
    public init(_ val: Int, _ left: TreeNode?, _ right: TreeNode?) {
        self.val = val
        self.left = left
        self.right = right
    }
}

解法：

func maxProduct(_ root: TreeNode?) -> Int {
    // 结果
    var res: Int = Int.min
    // 总和
    var allSum: Int = 0
    // 第一部分子树和
    var nodeSum: Int = 0
    
    // 求总和
    func sum(_ root: TreeNode?) -> Int {
        guard let r = root else { return 0 }
        return r.val + sum(r.left) + sum(r.right)
    }
    
    // 递归求子树和
    func dfs(_ root: TreeNode?) -> Int {
        guard let r = root else { return 0 }
        
        // 子树和
        nodeSum = r.val + dfs(r.left) + dfs(r.right)
        
        // 计算最大值
        res = max(res, (allSum - nodeSum) * nodeSum)
        return nodeSum
    }
    
    allSum = sum(root)
    _ = dfs(root)
    res = res % (Int(1e9) + 7)
    
    return res
}