代码随想录算法训练营第23天 | 669. 修剪二叉搜索树 + 108.将有序数组转换为二叉搜索树 + 538.把二叉搜索树转换为累加树

最新推荐文章于 2024-07-12 18:56:03 发布

HY.YUE

最新推荐文章于 2024-07-12 18:56:03 发布

阅读量77

点赞数 1

分类专栏： LeetCode 文章标签：算法 python leetcode

本文链接：https://blog.csdn.net/weixin_42286468/article/details/130993025

版权

LeetCode 专栏收录该内容

53 篇文章 5 订阅

订阅专栏

今日任务

669. 修剪二叉搜索树 - Medium

108.将有序数组转换为二叉搜索树 - Easy

538.把二叉搜索树转换为累加树 - Medium

总结

递归三部曲

题目汇总

669. 修剪二叉搜索树 - Medium

题目链接：力扣-669. 修剪二叉搜索树

给你二叉搜索树的根节点 root ，同时给定最小边界low 和最大边界 high。通过修剪二叉搜索树，使得所有节点的值在[low, high]中。修剪树不应该改变保留在树中的元素的相对结构 (即，如果没有被移除，原有的父代子代关系都应当保留)。可以证明，存在唯一的答案。

所以结果应当返回修剪好的二叉搜索树的新的根节点。注意，根节点可能会根据给定的边界发生改变。

提示：递归中再递归

# 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 trimBST(self, root: Optional[TreeNode], low: int, high: int) -> Optional[TreeNode]:
        if not root: return None
        if root.val < low:
            return self.trimBST(root.right, low, high)
        if root.val > high:
            return self.trimBST(root.left, low, high)
        root.left = self.trimBST(root.left, low, high)
        root.right = self.trimBST(root.right, low, high)
        return root

108.将有序数组转换为二叉搜索树 - Easy

题目链接：力扣-108. 将有序数组转换为二叉搜索树

给你一个整数数组 nums ，其中元素已经按升序排列，请你将其转换为一棵高度平衡二叉搜索树。

高度平衡二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 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
        left, right = 0, len(nums)-1
        mid = left + (right - left) // 2
        root = TreeNode(nums[mid])
        left_nums = nums[:mid]
        root.left = self.sortedArrayToBST(nums[:mid])
        root.right = self.sortedArrayToBST(nums[mid+1:])
        return root

538.把二叉搜索树转换为累加树 - Medium

题目链接：力扣-538. 把二叉搜索树转换为累加树

给出二叉搜索树的根节点，该树的节点值各不相同，请你将其转换为累加树（Greater Sum Tree），使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。

提示：反中序遍历，前位指针

# 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 __init__(self):
        self.pre = 0

    def convertBST(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
        self.tranversal(root)
        return root

    def tranversal(self, root):
        if not root: return
        self.tranversal(root.right)
        root.val += self.pre
        self.pre = root.val
        self.tranversal(root.left)