700 二叉搜索树中的搜索

最新推荐文章于 2024-05-21 20:29:42 发布

来颗花椰菜

最新推荐文章于 2024-05-21 20:29:42 发布

阅读量79

点赞数

分类专栏：二叉树文章标签：算法 leetcode 数据结构

本文链接：https://blog.csdn.net/qq_44658259/article/details/126794136

版权

二叉树专栏收录该内容

32 篇文章 0 订阅

订阅专栏

700 二叉搜索树中的搜索

给定二叉搜索树（BST）的根节点 root 和一个整数值 val。

你需要在 BST 中找到节点值等于 val 的节点。返回以该节点为根的子树。如果节点不存在，则返回 null 。

输入：root = [4,2,7,1,3], val = 2
输出：[2,1,3]

输入：root = [4,2,7,1,3], val = 5
输出：[]

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def searchBST(self, root: Optional[TreeNode], val: int) -> Optional[TreeNode]:
        # 非递归1
        # from collections import deque
        # que = deque([root])
        # while que:
        #     node = que.popleft()
        #     if node.val != val:
        #         if node.left:
        #             que.append(node.left)
        #         if node.right:
        #             que.append(node.right)
        #     else:
        #         return node
        # return 
        # 非递归2
        # while root:
        #     if root.val > val:
        #         root = root.left
        #     elif root.val < val:
        #         root = root.right
        #     else: return root
        # return 
        # 递归
        if not root or root.val == val:return root
        if root.val < val:
            return self.searchBST(root.right,val)
        if root.val > val:
            return self.searchBST(root.left, val)