【树-简单】700. 二叉搜索树中的搜索

最新推荐文章于 2024-08-05 20:00:21 发布

菜菜2022

最新推荐文章于 2024-08-05 20:00:21 发布

阅读量58

点赞数

分类专栏：刷题 # leetcode

本文链接：https://blog.csdn.net/kz_java/article/details/118269665

版权

leetcode 同时被 2 个专栏收录

714 篇文章 5 订阅

订阅专栏

刷题

643 篇文章 5 订阅

订阅专栏

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

例如，

给定二叉搜索树:

和值: 2

你应该返回如下子树:

  2     
 / \   
1   3

在上述示例中，如果要找的值是 5，但因为没有节点值为 5，我们应该返回 NULL。
【代码】
在这里插入图片描述

class Solution:
    def searchBST(self, root: TreeNode, val: int) -> TreeNode:
        if not root:
            return 
        que=[root]
        while que:
            node=que.pop(0)
            if node.val==val:
                return node
            if node.left:
                que.append(node.left)
            if node.right:
                que.append(node.right)
        return None

【方法2：借助BST性质提高性能】
在这里插入图片描述

class Solution:
    def searchBST(self, root: TreeNode, val: int) -> TreeNode:
        if not root:
            return None
        node=root
        while node:
            if not node:
                return node
            elif node.val==val:
                return node
            elif node.val>val:
                node=node.left
            elif node.val<val:
                node=node.right
        return node

【方法3：迭代优化版本】
在这里插入图片描述

class Solution:
    def searchBST(self, root: TreeNode, val: int) -> TreeNode:
        while root and root.val!=val:
            if root.val>val:
                root=root.left
            else:
                root=root.right
        return root

【方法4：递归优化版本】
在这里插入图片描述

class Solution:
    def searchBST(self, root: TreeNode, val: int) -> TreeNode:
        if root is None or val == root.val:
            return root
        
        return self.searchBST(root.left, val) if val < root.val else self.searchBST(root.right, val)