刷题笔记 | 将二叉搜索树变平衡

将二叉搜索树变平衡

二叉查找树（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)