Leetcode 669.修剪二叉搜索树
1 题目描述(Leetcode题目链接)
给定一个二叉搜索树,同时给定最小边界L 和最大边界 R。通过修剪二叉搜索树,使得所有节点的值在[L, R]中 (R>=L) 。你可能需要改变树的根节点,所以结果应当返回修剪好的二叉搜索树的新的根节点。
输入:
1
/ \
0 2
L = 1
R = 2
输出:
1
\
2
输入:
3
/ \
0 4
\
2
/
1
L = 1
R = 3
输出:
3
/
2
/
1
2 题解
根据二叉搜索树的性质,递归实现,
- 如果当前节点的值小于 L L L,则它的右孩子还有可能满足条件;
- 如果当前节点的值大于 R R R,则它的左孩子还有可能满足条件;
- 如果当前节点的值在 L , R L,R L,R之间,则左右孩子都可能满足条件 。
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def trimBST(self, root: TreeNode, L: int, R: int) -> TreeNode:
if not root:
return root
if root.val < L:
return self.trimBST(root.right, L, R)
if root.val > R:
return self.trimBST(root.left, L, R)
root.left = self.trimBST(root.left, L, R)
root.right = self.trimBST(root.right, L, R)
return root