算法训练营Day23(二叉树9)

669. 修剪二叉搜索树 力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台

class Solution:
    def trimBST(self, root: Optional[TreeNode], low: int, high: int) -> Optional[TreeNode]:
        if root is None:
            return None
        if root.val < low:
            # 寻找符合区间 [low, high] 的节点
            return self.trimBST(root.right, low, high)
        if root.val > high:
            # 寻找符合区间 [low, high] 的节点
            return self.trimBST(root.left, low, high)
        root.left = self.trimBST(root.left, low, high)  # root.left 接入符合条件的左孩子
        root.right = self.trimBST(root.right, low, high)  # root.right 接入符合条件的右孩子
        return root

108.将有序数组转换为二叉搜索树力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台

提醒

本题就简单一些，可以尝试先自己做做。

注意区间的定义

class Solution:
    def traversal(self, nums: List[int], left: int, right: int) -> TreeNode:
        if left > right:
            return None
        
        mid = left + (right - left) // 2
        root = TreeNode(nums[mid])
        root.left = self.traversal(nums, left, mid - 1)
        root.right = self.traversal(nums, mid + 1, right)
        return root
    #主函数
    def sortedArrayToBST(self, nums: List[int]) -> TreeNode:
        root = self.traversal(nums, 0, len(nums) - 1)
        return root

538.把二叉搜索树转换为累加树力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台

提醒

本题也不难，在 求二叉搜索树的最小绝对差 和 众数 那两道题目 都讲过了 双指针法，思路是一样的。

 注意：本题和 1038: 力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台 相同

思考

为什么pre是int型，而不是treenode

class Solution:
    def convertBST(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
        self.pre = 0  # 记录前一个节点的数值
        self.traversal(root)
        return root
    def traversal(self, cur):
        if cur is None:
            return        
        #右
        self.traversal(cur.right)
        #中
        cur.val += self.pre
        self.pre = cur.val
        #左
        self.traversal(cur.left)

总结篇  

1.碰到二叉搜索树就用中序遍历，其他题优先使用后序遍历

2.如果题目中有明显的“从根节点开始遍历”就用前序，如求和，路径