【二叉树】最小高度树

最新推荐文章于 2022-05-21 22:13:34 发布

豪冷啊

最新推荐文章于 2022-05-21 22:13:34 发布

阅读量1.5k

点赞数 1

分类专栏：算法文章标签：算法深度优先 swift ios objective-c

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

版权

算法专栏收录该内容

121 篇文章 0 订阅

订阅专栏

0x00 题目

给定一个 有序 整数数组
元素各不相同且按 升序 排列
编写一个算法，创建一棵 高度最小 的二叉 搜索 树

给定有序数组: [-10,-3,0,5,9]

一个可能的答案是：[0,-3,9,-10,null,5]
它可以表示下面这个高度平衡二叉搜索树：

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 sortedArrayToBST(_ nums: [Int]) -> TreeNode? {
    func build(_ nums: [Int], _ left: Int, _ right: Int) -> TreeNode? {
    
        // 左边 > 右边时，返回
        if left > right { return nil }
        
        // 找到中间的索引
        let mid = (left + right) / 2
        
        // 创建对应 节点
        let node = TreeNode(nums[mid])
        
        // 递归创建左节点
        node.left = build(nums, left, mid-1)
        
        // 递归创建右节点
        node.right = build(nums, mid+1, right)
        
        // 返回该节点
        return node
    }
    
    return build(nums, 0, nums.count-1)
}