LeetCode题669 Trim a Binary Search Tree

题目:给定一个二叉搜索树和最低和最高的边界LR, 对二叉搜索树进行修剪,使其所有元素都在[L, R](R> = L)之间。可以更改树的根,所以结果返回修剪后的二叉搜索树的新根。

LeetCode题669 Trim a Binary Search Tree

思路:先看根节点

1、根节点小于最低边界,则丢弃左子树和根,只保留右子树,再对右子树进行递归

2、根节点大于最高边界,则丢弃右子树和根,只保留左子树,再对左子树进行递归

3、根节点大于等于最低边界,小于或等于最高边界,则对二叉搜索树的左右子树进行递归

# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None


class Solution:
    def trimBST(self, root, L, R):
        if not root:
            return None
        if root.val < L:
            return self.trimBST(root.right,L,R)
        elif root.val >R:
            return self.trimBST(root.left,L,R)
        else:
            root.left=self.trimBST(root.left,L,R)
            root.right=self.trimBST(root.right,L,R)
        return root