- 将二叉搜索树变平衡
给你一棵二叉搜索树,请你返回一棵 平衡后 的二叉搜索树,新生成的树应该与原来的树有着相同的节点值。
如果一棵二叉搜索树中,每个节点的两棵子树高度差不超过 1 ,我们就称这棵二叉搜索树是 平衡的 。
如果有多种构造方法,请你返回任意一种。
示例:
输入:root = [1,null,2,null,3,null,4,null,null]
输出:[2,1,3,null,null,null,4]
解释:这不是唯一的正确答案,[3,1,4,null,2,null,null] 也是一个可行的构造方案。
提示:
树节点的数目在 1 到 10^4 之间。
树节点的值互不相同,且在 1 到 10^5 之间。
思路:
树的问题用的最多的就是递归求解:
先用中序遍历把原来的二叉树的元素放到vals数组里面,然后从下到上那个构造二叉平衡。
二叉平衡树的要求:左右子树的高度差不能 > 1
构造BST的时候,m妨碍中间,递归从下往上构建,书的问题一般都用递归,看代码
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def balanceBST(self, root: TreeNode) -> TreeNode:
vals = []
def inorder(root: TreeNode):
if not root:
return None
inorder(root.left)
vals.append(root.val)
inorder(root.right)
def build(l: int, r: int): # 构建从L到R的BST
if l > r:
return None
m = int(l + (r-l)/2) # 因为m的坐标是中间位置,可以保证BSt的性质
root = TreeNode(vals[m])
root.left = build(l, m-1)
root.right = build(m+1, r)
return root
inorder(root)
return build(0, len(vals)-1)