1. 程式人生 > >leetcode108python將有序陣列轉換為二叉搜尋樹

leetcode108python將有序陣列轉換為二叉搜尋樹

將一個按照升序排列的有序陣列,轉換為一棵高度平衡二叉搜尋樹。

本題中,一個高度平衡二叉樹是指一個二叉樹每個節點 的左右兩個子樹的高度差的絕對值不超過 1。

示例:

給定有序陣列: [-10,-3,0,5,9],

一個可能的答案是:[0,-3,9,-10,null,5],它可以表示下面這個高度平衡二叉搜尋樹:

      0
     / \
   -3   9
   /   /
 -10  5
python3
# Definition for a binary tree node.
# 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
        mid=len(nums)//2#找到中間節點
        root=TreeNode(nums[mid])#當前節點為根節點
        root.left=self.sortedArrayToBST(nums[:mid])#小於當前根節點的作為左子樹
        root.right=self.sortedArrayToBST(nums[mid+1:])#大於當前根節點的作為右子樹
        return root