Update bst_search.py

This commit is contained in:
marina 2023-08-03 13:31:57 -07:00 committed by GitHub
parent 419eeee0dc
commit fb8081ae42
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -2,14 +2,31 @@
# -*- coding: utf-8 -*-
# author: bt3gl
def search_bst(root, val):
def search_bst_recursive(root, val):
if root is None or root.val == val:
return root
if val > root.val:
return search_bst(root.right, val)
return search_bst_recursive(root.right, val)
else:
return search_bst(root.left, val)
return search_bst_recursive(root.left, val)
def search_bst_iterative(root, val):
node = root
while node:
if node.val == val:
return node
if node.val < val:
node = node.right
else:
node = node.left
return False