LeetCode--108--將有序數組轉化為二叉搜索樹
阿新 • • 發佈:2018-09-09
tree leet solution eno ron bsp 構造 pos int
問題描述:
將一個按照升序排列的有序數組,轉換為一棵高度平衡二叉搜索樹。
本題中,一個高度平衡二叉樹是指一個二叉樹每個節點 的左右兩個子樹的高度差的絕對值不超過 1。
示例:
給定有序數組: [-10,-3,0,5,9], 一個可能的答案是:[0,-3,9,-10,null,5],它可以表示下面這個高度平衡二叉搜索樹: 0 / -3 9 / / -10 5
思路:由於數組是有序數組,所以只需用二分法構造即可,mid即為根節點,0:mid 的中間點為左子樹的根節點,mid+1:末尾的中間點為右子樹的根節點,遞歸的構造即可。
方法1:
1 class Solution(object):2 def sortedArrayToBST(self, nums): 3 """ 4 :type nums: List[int] 5 :rtype: TreeNode 6 """ 7 if nums: 8 midPos = len(nums) // 2 9 mid = nums[midPos] 10 root = TreeNode(mid) 11 root.left = self.sortedArrayToBST(nums[:midPos])12 root.right = self.sortedArrayToBST(nums[midPos+1:]) 13 return root
2018-09-09 16:00:01
LeetCode--108--將有序數組轉化為二叉搜索樹