Python - árvore de pesquisa
Uma árvore de pesquisa binária (BST) é uma árvore na qual todos os nós seguem as propriedades mencionadas abaixo - A subárvore esquerda de um nó tem uma chave menor ou igual à chave do nó pai. A subárvore direita de um nó possui uma chave maior do que a chave de seu nó pai. Assim, o BST divide todas as suas subárvores em dois segmentos; a subárvore esquerda e a subárvore direita e podem ser definidas como -
left_subtree (keys) ≤ node (key) ≤ right_subtree (keys)
Pesquise um valor em uma árvore B
A pesquisa de um valor em uma árvore envolve a comparação do valor recebido com os nós de saída do valor. Aqui também percorremos os nós da esquerda para a direita e, finalmente, com o pai. Se o valor procurado não corresponder a nenhum valor exitign, retornaremos a mensagem não encontrada, caso contrário, a mensagem encontrada será retornada.
class Node:
def __init__(self, data):
self.left = None
self.right = None
self.data = data
# Insert method to create nodes
def insert(self, data):
if self.data:
if data < self.data:
if self.left is None:
self.left = Node(data)
else:
self.left.insert(data)
elif data > self.data:
if self.right is None:
self.right = Node(data)
else:
self.right.insert(data)
else:
self.data = data
# findval method to compare the value with nodes
def findval(self, lkpval):
if lkpval < self.data:
if self.left is None:
return str(lkpval)+" Not Found"
return self.left.findval(lkpval)
elif lkpval > self.data:
if self.right is None:
return str(lkpval)+" Not Found"
return self.right.findval(lkpval)
else:
print(str(self.data) + ' is found')
# Print the tree
def PrintTree(self):
if self.left:
self.left.PrintTree()
print( self.data),
if self.right:
self.right.PrintTree()
root = Node(12)
root.insert(6)
root.insert(14)
root.insert(3)
print(root.findval(7))
print(root.findval(14))
Quando o código acima é executado, ele produz o seguinte resultado -
7 Not Found
14 is found