leetcode108python將有序陣列轉換為二叉搜尋樹
阿新 • • 發佈:2018-12-24
將一個按照升序排列的有序陣列,轉換為一棵高度平衡二叉搜尋樹。
本題中,一個高度平衡二叉樹是指一個二叉樹每個節點 的左右兩個子樹的高度差的絕對值不超過 1。
示例:
給定有序陣列: [-10,-3,0,5,9], 一個可能的答案是:[0,-3,9,-10,null,5],它可以表示下面這個高度平衡二叉搜尋樹: 0 / \ -3 9 / / -10 5python3
# 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