给定一个二叉搜索树,同时给定最小边界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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/trim-a-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
第一种思路:
中序遍历得到升序数组,然后只保留在区间范围的值,再重新构造二叉搜索树。
类似LeetCode-Python-108. 将有序数组转换为二叉搜索树
时间复杂度:O(N)
空间复杂度:O(N)
第二种思路:
递归处理每个节点,分情况讨论:
1. 如果node.val在区间内,则说明该node需要保留,继续递归处理左右子节点
2. 如果node.val < L, 说明node和node.left 都无需保留
3. 如果node.val > R, 说明node和node.right 都无需保留
时间复杂度:O(N)
空间复杂度:O(N)
class Solution(object):
def trimBST(self, root, L, R):
"""
:type root: TreeNode
:type L: int
:type R: int
:rtype: TreeNode
"""
if not root:
return None
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