题目分析:
将一个按照升序排列的有序数组,转换为一棵高度平衡二叉搜索树。本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定有序数组: [-10,-3,0,5,9],一个可能的答案是:[0,-3,9,-10,null,5],它可以表示下面这个高度平衡二叉搜索树:
解题思路:
这一题利用二分查找与递归的思想解决,因为是把一个有序的数组转换成二叉搜索树,所以
- 数组的中间元素就是二叉搜索树的根节点
- 数组中间元素左侧元素是左子树
- 数组中间元素右侧元素是右子树
- 递归重复1,2,3步骤
- 注:这个过程其实也就是中序遍历逆过程
提交代码:(递归,Runtime: 56 ms, faster than 99.23% )
class Solution:
def sortedArrayToBST(self, nums: 'List[int]') -> TreeNode:
if nums == []: return None
index = len(nums) // 2
root = TreeNode(nums[index])
root.left = self.sortedArrayToBST(nums[:index])
root.right = self.sortedArrayToBST(nums[index + 1:])
return root