leetcode-1382. 将二叉搜索树变平衡

  1. 将二叉搜索树变平衡
    给你一棵二叉搜索树,请你返回一棵 平衡后 的二叉搜索树,新生成的树应该与原来的树有着相同的节点值。

如果一棵二叉搜索树中,每个节点的两棵子树高度差不超过 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)
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值