leetcode-669. 修剪二叉搜索树

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

示例 1:

输入:
1
/
0 2

L = 1
R = 2

输出:
1

2
示例 2:

输入:
3
/
0 4

2
/
1

L = 1
R = 3

输出:
3
/
2
/
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 trimBST(self, root: TreeNode, low: int, high: int) -> TreeNode:
        # 思路：因为是二叉搜索树，所以只有三种情况
        if not root:
            return 
        if low <= root.val <= high:
            self.trimBST(root.left, low, high)
            self.trimBST(root.right, low, high)
        elif root.val < low:
            root.left = None
            self.trimBST(root.right, low, high)
        elif root.val > high:
            root.right = None
            self.trimBST(root.left, low, high)
        return root