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
解题思路:
1.首先要先明白二叉搜索树是什么?具有什么性质?
1)首先二叉搜索树是一个二叉树,每个节点都可能具有左右子树?当然也可能没有,name就是叶子节点。
2)二叉搜索树的基本性质:
1,任意节点x,其左子树中的key不大于x.key,其右子树中的key不小于x.key。
2,不同的二叉搜索树可以代表同一组值的集合。
3,二叉搜索树的基本操作和树的高度成正比,所以如果是一棵完全二叉树的话最坏运行时间为Θ(lgn),但是若是一个n个节点连接成的线性树,那么最坏运行时间是Θ(n)。
4,根节点是唯一一个parent指针指向NIL节点的节点。
5,每一个节点至少包括key、left、right与parent四个属性,构建二叉搜索树时,必须存在针对key的比较算法。
2.回到本题,所以,我们如果发现一个节点的值比L小,那么它的左子树的所有节点的值肯定都比这个值小,所以直接去掉。如果我们发现一个节点的值比R大,那么它的所有右子树的节点的值都比它大,也都可以直接去掉。
3.最后我们用递归就可以实现这个题的解了。
代码实现:
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def trimBST(self, root, L, R):
"""
:type root: TreeNode
:type L: int
:type R: int
:rtype: TreeNode
"""
if root is None:
return root
elif root.val>R:
return self.trimBST(root.left, L, R)
elif root.val<L:
return self.trimBST(root.right, L, R)
root.left = self.trimBST(root.left, L, R)
root.right = self.trimBST(root.right, L, R)
return root