【二叉树】有序数组转换为二叉搜索树

0x00 题目

给你一个整数数组 nums ,其中元素已经按【升序】排列,请你将其转换为一棵 【高度平衡】二叉搜索树

高度平衡二叉树,是一棵满足每个节点的左右两个子树的【高度差】的绝对值不超过 1 的二叉树


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)
}

五笔学习

小程序:XWubi
配合公众号文章一起食用,效果更佳~
五笔教程之字根解析(二)
请添加图片描述


猜你喜欢

转载自blog.csdn.net/xjh093/article/details/121373707