669. Trim a Binary Search Tree

题目链接: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)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值