将二叉搜索树变平衡
二叉查找树(Binary Search Tree)
(又:二叉搜索树,二叉排序树)它或者是一棵空树,或者是具有下列性质的二叉树:
- 若它的左子树不空,则左子树上所有结点的值均小于它的根结点的值;
- 若它的右子树不空,则右子树上所有结点的值均大于它的根结点的值;
- 它的左、右子树也分别为二叉排序树。
题目描述
-
给你一棵二叉搜索树,请你返回一棵平衡后的二叉搜索树,新生成的树应该与原来的树有着相同的节点值。
-
如果一棵二叉搜索树中,每个节点的两棵子树高度差不超过 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
之间。
解题思路
- 首先通过中序遍历得到有序数组(二叉搜索树的中序遍历是一个有序数组),然后将这个有序数组转化为平衡二叉搜索树。
- 转化:将数组中间的元素作为根节点,左边的元素作为左子树,右边的元素作为右子树,递归下去即可。
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def balanceBST(self, root):
"""
:type root: TreeNode
:rtype: TreeNode
"""
res = []
def inOrder(node):
if node:
inOrder(node.left)
res.append(node.val)
inOrder(node.right)
inOrder(root)
def sortedToBST(l, r):
if l > r: return None
mid = (l + r) >> 1
root = TreeNode(res[mid])
root.left = sortedToBST(l, mid - 1)
root.right = sortedToBST(mid + 1, r)
return root
return sortedToBST(0, len(res)-1)