108. 将有序数组转换为二叉搜索树

题目

给你一个整数数组 nums ,其中元素已经按 升序 排列,请你将其转换为一棵 高度平衡 二叉搜索树。
高度平衡 二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。
示例 1:
108. 将有序数组转换为二叉搜索树 - 图1

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

108. 将有序数组转换为二叉搜索树 - 图2
示例 2:
108. 将有序数组转换为二叉搜索树 - 图3

输入:nums = [1,3]
输出:[3,1]
解释:[1,3] 和 [3,1] 都是高度平衡二叉搜索树。

提示:

  • 1 <= nums.length <= 104
  • 104 <= nums[i] <= 104
  • nums严格递增 顺序排列

题解

就是简单的分治即可

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def sortedArrayToBST(self, nums: List[int]) -> TreeNode:
        n=len(nums)
        if n==0:
            return None
        elif n==1:
            return TreeNode(nums[0])
        else:
            i=n//2
            left=self.sortedArrayToBST(nums[:i])
            right=self.sortedArrayToBST(nums[i+1:])
            return TreeNode(nums[i],left,right)

108. 将有序数组转换为二叉搜索树 - 图4