题目:
给你一个整数数组 nums
,其中元素已经按 升序 排列,请你将其转换为一棵 高度平衡 二叉搜索树。
高度平衡 二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。
答案:
给定一个有序数组,将其转化为树,也就是说这个有序数组是二叉树的中序遍历,但是仅凭中序遍历我们无法确定二叉树,加上高度平衡,答案不唯一,但是范围很小。总的来说,有三种方法:
如果中序遍历为奇数,那么我们只能选择最中间的数字作为根节点,再之后递归地进行选择根节点。
如果遍历序列为偶数,就会有三种选择,mid的左边和右边以及任意选择一个节点,之后同样递归选择
需注意,当left>right时,该二叉树为空。
class Solution:
def sortedArrayToBST(self, nums: List[int]) -> Optional[TreeNode]:
def CT(left, right):
if left > right:
return None
mid = (left + right) // 2
root = TreeNode(nums[mid])
root.left = CT(left, mid - 1)
root.right = CT(mid + 1, right)
return root
return CT(0, len(nums) - 1)