【二叉树】最大二叉树

原创

haocold 2022-10-24 20:05:01 博主文章分类：算法 ©著作权

©著作权归作者所有：来自51CTO博客作者haocold的原创作品，请联系作者获取转载授权，否则将追究法律责任

0x00 题目

给定一个不重复的整数数组 nums 构建一个 最大 二叉树
最大二叉树可以用下面的算法从 nums 递归地构建:
创建一个 根节点，其值为 nums 中的最大值
递归地在最大值 左边 的子数组前缀上构建 左子树 递归地在最大值 右边 的子数组后缀上构建 右子树 返回 nums 构建的最大二叉树

0x01 思路

找到数组中的 最大数 构建 根节点 根据索引把数组 一分为二 对两个数组进行 递归 操作

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 constructMaximumBinaryTree(_ nums: [Int]) -> TreeNode? {
    if nums.isEmpty { return nil }
    
    // 查找数组中最大数的索引
    var index: Int = 0
    for i in 0..<nums.count {
        if nums[index] < nums[i] {
            index = i
        }
    }
    
    // 构建根节点
    let root = TreeNode(nums[index])
    // 递归构建左子树
    root.left = constructMaximumBinaryTree(Array(nums[0..<index]))
    // 递归构建右子树
    root.right = constructMaximumBinaryTree(Array(nums[(index+1)...]))
    
    return root
}