题目链接:https://leetcode.com/problems/trim-a-binary-search-tree/description/
题目要求把不符合要求的节点”剪掉”即可。这是一颗二叉搜索树,需要利用二叉搜索树的性质。实现起来有点麻烦。过程也不太好叙述啊[衰]
1、当前根节点满足[L,R]
1.1、则检查当前根节点左子树的值
1.2、检查当前根节点的右子树的值
1.3、递归检查左子树
1.4、递归检查右子树
2、当前根节点小于L
检查右子树上值
3、当前根节点大于R
检查左子树上值
代码实现:
# -*- coding:utf-8 -*-
# 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 inOrder(self,root,L,R):
if root:
if root.val>=L and root.val<=R:#本节点存在且符合条件
if root.left and root.left.val<L:#左子树存在,且左子树的值小于下限,那么其左子树上的值也一定小于下限
# root.left.left=None
root.left=root.left.right#其右子树上的值可能大于下限,然后递归检查
if root.right and root.right.val>R:#右子树存在且右子树的值大于上限,那么其右子树上的值也一定大于上限
# root.right.right=None
root.right=root.right.left#其左子树上的值可能小于下限,需要递归检查
#经过上面两个if的处理,root的两个左右子树都是可能满足条件的节点,需要继续递归检查
root.left=self.inOrder(root.left,L,R)#一开始没有写 root.left=,只是递归调用,导致结果出错
root.right=self.inOrder(root.right,L,R)
return root
#以下是针对“根节点”本身就不符合条件的情况
elif root.val<L:#“根节点”小于下限,那么可能的“根节点”在右子树上
return self.inOrder(root.right,L,R)
elif root.val>R:#“根节点”大于上限,那么可能的“根节点”在左子树上
return self.inOrder(root.left,L,R)
return None#空节点则直接返回None
def trimBST(self, root, L, R):
"""
:type root: TreeNode
:type L: int
:type R: int
:rtype: TreeNode
"""
return self.inOrder(root,L,R)