Leetcode刷题Python108. 将有序数组转换为二叉搜索树

Posted 2022-08-10 Better Bench

1 题目

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

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

示例 1：

输入：nums = [-10,-3,0,5,9]
输出：[0,-3,9,-10,null,5]
解释：[0,-10,5,null,-3,null,9] 也将被视为正确答案：

2 解析

递归生成数，以前序遍历的方式生成树

3 Python实现

class Solution:
    def sortedArrayToBST(self, nums: List[int]) -> Optional[TreeNode]:

        mid = len(nums)//2
        if nums:
            root = TreeNode(nums[mid])
            root.left = self.sortedArrayToBST(nums[:mid])
            root.right = self.sortedArrayToBST(nums[mid+1:])
            return root