gpt4 book ai didi

700. Search in a Binary Search Tree 二叉搜索树中的搜索

转载 作者:大佬之路 更新时间:2024-01-31 14:17:54 27 4
gpt4 key购买 nike

题目地址:https://leetcode.com/problems/search-in-a-binary-search-tree/description/

题目描述

Given the root node of a binary search tree (BST) and a value. You need to find the node in the BST that the node's value equals the given value. Return the subtree rooted with that node. If such node doesn't exist, you should return NULL.

Forexample,

Given the tree:

    4
   / \
  2   7
 / \
1   3

And the value to search: 2

Youshould return this subtree:

  2     
 / \   
1   3

Inthe example above, if we want to search the value 5, since there is no node with value 5, we should return NULL.

题目大意

在一个BST中查找某值,如果找到的话,返回找到的那个节点,如果找不到就返回None.

解题方法

递归

只要是BST查找,那肯定都好说。树的查找比较简单的就是递归,如果节点是空,那么肯定没找到;节点值相等,返回这个节点;如果节点值小于要查找的值,那么在当前节点的右子树中找;否则在左子树中找。

代码如下:

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def searchBST(self, root, val):
        """
        :type root: TreeNode
        :type val: int
        :rtype: TreeNode
        """
        if not root:
            return None
        if root.val == val:
            return root
        elif root.val < val:
            return self.searchBST(root.right, val)
        else:
            return self.searchBST(root.left, val)

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22

DDKK.COM 弟弟快看-教程,程序员编程资料站,版权归原作者所有

本文经作者:负雪明烛 授权发布,任何组织或个人未经作者授权不得转发

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com