1. 程式人生 > >LeetCode--108--將有序數組轉化為二叉搜索樹

LeetCode--108--將有序數組轉化為二叉搜索樹

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--將有序數組轉化為二叉搜索樹