力扣—108有序数组转换为二叉搜索树

题目描述

将一个按照升序排列的有序数组，转换为一棵高度平衡二叉搜索树。

本题中，一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。

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

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

      0
     / \
   -3   9
   /   /
 -10  5

解题思路

基于二分法建立二叉树，二分法可以使左右节点数相差最多为1

class TreeNode(object):
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

class Solution(object):
    def sortedArrayToBST(self, nums):
        """
        :type nums: List[int]
        :rtype: TreeNode
        """
        if not nums:
            return None
        root = self.Build_tree(nums, 0, len(nums))
        print(self.Print_tree(root))

    def Print_tree(self, root):
        if not root:
            return []
        return self.Print_tree(root.left) + [root.val] + self.Print_tree(root.right)

    def Build_tree(self, nums, start, end):
        if start >= end:
            return None
        index = (start + end) // 2
        root = TreeNode(nums[index])
        root.left = self.Build_tree(nums, start, index)
        root.right = self.Build_tree(nums, index + 1, end)
        return root

S = Solution()
print(S.sortedArrayToBST([-10,-3,0,5,9]))