【LeetCode】第108题：将有序数组转换为二叉搜索树

最新推荐文章于 2024-03-22 06:02:20 发布

inganxu

最新推荐文章于 2024-03-22 06:02:20 发布

阅读量333

点赞数

分类专栏： LeetCode练习笔记文章标签： leetcode 数据结构算法二叉树经验分享

本文链接：https://blog.youkuaiyun.com/weixin_46159679/article/details/121778683

版权

LeetCode练习笔记专栏收录该内容

8 篇文章

订阅专栏

题目：

给定一个升序排列的数组，请将转换为一颗高度平衡的二叉搜索树

每个节点的左右两个子树的高度差的绝对值不超过1

题目链接：将有序数组转换为二叉搜索树

题目限制：

nums按严格递增顺序排列

子节点高度差不超过1

解题思路：

先判断数组是否有元素

找到数组最中间的数，以该数作为根节点

左边的元素就是根节点的左子节点

右边的元素就是根节点的右子节点

根节点向左移动一位，再以该节点作为根节点，左边的元素.....，右边的元素......

根节点的右边也一样，迭代遍历

整个过程其实不需要知道数组中的数字有多大，因为全程都是通过索引进行排序

实现代码：

class Solution:
    def sortedArrayToBST(self, nums: List[int]) -> TreeNode:
        if not nums:
            return None

        val = len(nums) // 2                        # 找到数组最中间的元素索引位置
        node = TreeNode(nums[val])
        left = nums[:val]                           # 根节点的左子节点
        right = nums[val + 1:]                      # 根节点的右子节点
        node.left = self.sortedArrayToBST(left)     # 遍历迭代
        node.right = self.sortedArrayToBST(right)

        return node