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

669. 修剪二叉搜索树

题目：给定一个二叉搜索树，同时给定最小边界L 和最大边界 R。通过修剪二叉搜索树，使得所有节点的值在[L, R]中 (R>=L) 。你可能需要改变树的根节点，所以结果应当返回修剪好的二叉搜索树的新的根节点。

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

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)
        
        if low <= root.val <= high:
            root.left = self.trimBST(root.left, low, high) # 接入符合条件的左孩子
            root.right = self.trimBST(root.right, low, high) # 接入符合条件的右孩子
            return root

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

题目：将一个按照升序排列的有序数组，转换为一棵高度平衡二叉搜索树。

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

class Solution:
    def sortedArrayToBST(self, nums: List[int]) -> Optional[TreeNode]:
        root = self.inversal(nums, 0, len(nums) - 1)
        return root

    def inversal(self, nums: List[int], left: int, right: int) -> TreeNode:
        if left > right:
            return None
        # 选取中间元素作为分割点
        mid = left + (right - left) // 2 
        mid_root = TreeNode(nums[mid])

        mid_root.left = self.inversal(nums, left, mid - 1)
        mid_root.right = self.inversal(nums, mid + 1, right)
        return mid_root

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

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

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

class Solution:
    def __init__(self):
        self.pre = TreeNode()

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

    def traversal(self, root: TreeNode) -> None:
        if not root:
            return None
        # 单层递归逻辑：中序遍历的反译 - 右中左
        self.traversal(root.right)
        root.val += self.pre.val # root值加上pre的值
        self.pre = root  # pre 在后面“追着”root跑
        self.traversal(root.left)