代码随想录训练营Day 20|Python|Leetcode|669. 修剪二叉搜索树，108.将有序数组转换为二叉搜索树，538.把二叉搜索树转换为累加树-CSDN博客

本文链接：https://blog.csdn.net/qq_60959261/article/details/137648669

669. 修剪二叉搜索树

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

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

解题思路：

本体需要判断当要删除节点时，该节点的左右子树是否也应该本删除，因此需要再进行一次递归遍历

确认输入：def trim(root, low, high)

停止条件：if not root: return None; if root.val<low: 判断其右子树是否需要被删除，递归遍历，返回right；if root.val>high: 判断其左子树是否需要被删除，递归遍历，返回left

递归逻辑：使用root.left=trim(root.left, low, high)接住左边返回值，如果发现要删除的节点，会自动返回其left/right满足条件的值，root.right同理。

# 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:
            #run the recursive again
            right = self.trimBST(root.right, low, high)
            return right#let the previous step get the returned value and skip the wrong root
        elif root.val > high:
            left = self.trimBST(root.left, low, high)
            return left
        root.left = self.trimBST(root.left, low, high)
        root.right = self.trimBST(root.right, low, high)
        return root

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

给你一个整数数组 nums ，其中元素已经按升序排列，请你将其转换为一棵

平衡二叉搜索树。

解题思路：

由于是二叉搜索树并且nums已经按升序排列，平衡二叉树则可得叶子节点高度差不超过1，因此只要将中间节点作为root，并以此进行递归，可保证左右平衡。

输入参数：def sortedArrayToBST(nums)

停止条件：if not nums: return None

递归逻辑：root.left = self.sortedArrayToBST(left_side), root.right = self.sortedArrayToBST(root.right)

# 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
        idx = len(nums)//2
        root = TreeNode(nums[idx])
        left_side = nums[:idx]
        right_side = nums[idx+1:]#split and do recursive again
        root.left = self.sortedArrayToBST(left_side)
        root.right = self.sortedArrayToBST(right_side)
        return root

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

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

提醒一下，二叉搜索树满足下列约束条件：

节点的左子树仅包含键小于节点键的节点。
节点的右子树仅包含键大于节点键的节点。
左右子树也必须是二叉搜索树。

注意：本题和 1038: . - 力扣（LeetCode）相同

解题思路：

本题使用倒叙中序遍历，本来是左中右，改为右中左，因为要从大到小遍历数字叠加。

输入参数：def convertBST(cur)

停止条件：if not cur: return None

递归逻辑：初设pre = 0，convertBST(cur.right)，cur.val += pre, pre = cur.val，convertBST(cur.left)

# 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 convertBST(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
        pre = 0
        def dfs(cur):
            nonlocal pre
            if not cur:
                return None
            dfs(cur.right)
            cur.val += pre
            pre = cur.val
            dfs(cur.left)
        dfs(root)
        return root