108. 将有序数组转换为二叉搜索树
原始题目链接:https://leetcode-cn.com/problems/convert-sorted-array-to-binary-search-tree/
给你一个整数数组 nums ,其中元素已经按 升序 排列,请你将其转换为一棵 高度平衡 二叉搜索树。
高度平衡 二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。
解题思路:
构建树的过程就是递归的过程,给的数组有序要利用起来,题目要求构建二叉平衡搜索树,有两个要求,第一个是左右子树的高度差不能大于1,第二个是搜索树,左子树的节点都小于根节点,右子树的节点都大于根节点。构建的树不唯一,所以方便的做法是取中值作为根节点,再递归的构建树。
代码实现:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def sortedArrayToBST(self, nums: List[int]) -> Optional[TreeNode]:
if not nums:
return None
# 构建平衡二叉搜索树要求:1.中序遍历有序;2.高度差不超过1
# 因为给的数组是有序的,最简单的方式实现上面两个要求是取数组的中点为根,再构建树
mid = 0 + ((len(nums) - 0) // 2)
root = TreeNode(nums[mid])
root.left = self.sortedArrayToBST(nums[:mid])
root.right = self.sortedArrayToBST(nums[mid + 1:])
return root
参考文献:
https://leetcode-cn.com/problems/balance-a-binary-search-tree/solution/ping-heng-er-cha-shu-zhuan-ti-by-fe-lucifer-4/