首页 > 其他 > 详细

700. Search in a Binary Search Tree

时间:2019-03-31 22:03:47      阅读:140      评论:0      收藏:0      [点我收藏+]

700. Search in a Binary Search Tree 技术分享图片

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

例如

给定二叉搜索树:
        4
       /       2   7
     /     1   3

和值:2

你应该返回如下子树:
      2 
     /     1   3

在上述示例中,如果要找的值是5,但因为没有节点值为5,我们应该返回NULL。
# 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):
        """700. Search in a binary search tree

        :type root: TreeNode
        :type val: int
        :rtyoe: TreeNode
        """
        current_node = root
        while current_node != None:
            if current_node.val == val:
                return current_node
            elif current_node.val < val:
                current_node = current_node.right
            else:
                current_node = current_node.left
        return None

700. Search in a Binary Search Tree

原文:https://www.cnblogs.com/mrjoker-lzh/p/10633333.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!