將有序數組轉換成二叉搜索樹 （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)))