68. Leetcode 669. 修剪二叉搜索树 (二叉搜索树-基本操作类)

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

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

 
示例 1：

输入：root = [1,0,2], low = 1, high = 2
输出：[1,null,2]


示例 2：

输入：root = [3,0,4,null,2,null,null,1], low = 1, high = 3
输出：[3,2,null,1]


# 方法一 递归 思路

如果当前结点的值小于low，则递归右子树，并返回右子树符合条件的头结点；
如果当前结点的值大于high，则递归左子树，并返回左子树符合条件的头结点。
其目的是想要找到符合区间范围的二叉树节点。

if root.val < low:
    right = self.trimBST(root.right, low, high)
    return right

if root.val > high:
    left = self.trimBST(root.left, low, high)
    return left

将下一层处理完左子树的结果赋给root.left，处理完右子树的结果赋给root.right。最后返回root节点。

root.left = self.trimBST(root.left, low, high)
root.right = self.trimBST(root.right, low, high)

return root



# 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 root is None:
            return None
        
        if root.val < low:
            right = self.trimBST(root.right, low, high)
            return right
        
        if 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



# 方法二 迭代法 思路


迭代法：

在进行二叉树剪枝的时候，可以分为三步：

1. 将root移动到[low, high]范围内，左闭右闭区间；

while root and (root.val < low or root.val > high):
    if root.val < low:
        root = root.right
    else:
        root = root.left

2. 剪枝左子树：

cur = root
while cur:
    while cur.left and cur.left.val < low:
        cur.left = cur.left.right
    cur = cur.left

3. 剪枝右子树：

cur = root
while cur:
    while cur.right and cur.right.val > high:
        cur.right = cur.right.left
    cur = cur.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 root
        while root and (root.val < low or root.val > high):
            if root.val < low:
                root = root.right
            else:
                root = root.left
            
        cur = root
        while cur:
            while cur.left and cur.left.val < low:
                cur.left = cur.left.right
            cur = cur.left
        
        cur = root
        while cur:
            while cur.right and cur.right.val > high:
                cur.right = cur.right.left
            cur = cur.right

        return root