LeetCode-Python-700. 二叉搜索树中的搜索

最新推荐文章于 2022-10-24 18:18:33 发布

暴躁老哥在线刷题

最新推荐文章于 2022-10-24 18:18:33 发布

阅读量637

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode 文章标签： BST

本文链接：https://blog.youkuaiyun.com/qq_32424059/article/details/87381092

Leetcode 同时被 2 个专栏收录

821 篇文章

订阅专栏

Python

664 篇文章

订阅专栏

本文介绍了一种在二叉搜索树中查找特定值节点的方法。通过递归算法，根据BST特性，在左或右子树中搜索目标值，直至找到或返回NULL。示例展示了如何寻找值为2的节点，并解释了若目标值不存在于树中则返回NULL的情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给定二叉搜索树（BST）的根节点和一个值。你需要在BST中找到节点值等于给定值的节点。返回以该节点为根的子树。如果节点不存在，则返回 NULL。

例如，

给定二叉搜索树:

        4
       / \
      2   7
     / \
    1   3

和值: 2

你应该返回如下子树:

      2     
     / \   
    1   3

在上述示例中，如果要找的值是 5，但因为没有节点值为 5，我们应该返回 NULL。

思路：

简单的递归就可以了，

根据二叉搜索树(BST)的性质，如果当前节点值比目标值大，则在其左子树内搜索，

如果当前节点值比目标值小，则在其右子树内搜索。

# 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 searchBST(self, root, val):
        """
        :type root: TreeNode
        :type val: int
        :rtype: TreeNode
        """
        # print root.val
        if not root:
            return None
        if root.val == val:
            return root
        elif root.val > val:
            return self.searchBST(root.left, val)
        else:
            return self.searchBST(root.right, val)