将有序数组转换成二叉搜索树​​​​​​​（python）

LeetCode： 108 将有序数组转换成二叉搜索树


将一个按照 升序排列的有序数组
转换为一棵高度 平衡二叉搜索树
本题中，一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定有序数组: [-10,-3,0,5,9],
一个可能的答案是：[0,-3,9,-10,null,5]，它可以表示下面这个高度平衡二叉搜索树：
      0
     / \
   -3   9
   /   /
-10  5

思路：递归，每次都取有序数组的中值==root, 左侧为左子树递归+右侧为右子树递归

定义二叉树：

# Definition for a binary tree node.
class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

代码实现：

from typing import List
class Solution:
    def sortedArrayToBST(self, nums: List[int]) -> TreeNode:
        length=len(nums)
        if length==0:   return None
        if length==1:   return TreeNode(nums[0])
        else:
            root=TreeNode(nums[length//2])
            root.left=self.sortedArrayToBST(nums[:length//2])
            root.right=self.sortedArrayToBST((nums[length//2+1:]))
            return root

测试代码：

if __name__=="__main__":
    s = Solution()
    nums=[-10, -3, 0, 5, 9]
    r= s.sortedArrayToBST(nums)
    print(list(set(r)))