Add Binary Search Tree in data_structures

This commit is contained in:
Francisco Matias 2017-06-20 02:31:03 -03:00
parent 65de84b0f1
commit 4a9405e1b0
2 changed files with 102 additions and 0 deletions

1
.gitignore vendored
View File

@ -88,3 +88,4 @@ ENV/
# Rope project settings # Rope project settings
.ropeproject .ropeproject
.idea .idea
.DS_Store

View File

@ -0,0 +1,101 @@
'''
A binary search Tree
'''
class Node:
def __init__(self, label):
self.label = label
self.left = None
self.rigt = None
def getLabel(self):
return self.label
def setLabel(self, label):
self.label = label
def getLeft(self):
return self.left
def setLeft(self, left):
self.left = left
def getRight(self):
return self.rigt
def setRight(self, right):
self.rigt = right
class BinarySearchTree:
def __init__(self):
self.root = None
def insert(self, label):
#Create a new Node
node = Node(label)
if self.empty():
self.root = node
else:
dad_node = None
curr_node = self.root
while True:
if curr_node != None:
dad_node = curr_node
if node.getLabel() < curr_node.getLabel():
curr_node = curr_node.getLeft()
else:
curr_node = curr_node.getRight()
else:
if node.getLabel() < dad_node.getLabel():
dad_node.setLeft(node)
else:
dad_node.setRight(node)
break
def empty(self):
if self.root == None:
return True
return False
def preShow(self, curr_node):
if curr_node != None:
print(curr_node.getLabel())
print ('\n')
self.preShow(curr_node.getLeft())
self.preShow(curr_node.getRight())
def getRoot(self):
return self.root
'''
Example
8
/ \
3 10
/ \ \
1 6 14
/ \ /
4 7 13
'''
t = BinarySearchTree()
t.insert(8)
t.insert(3)
t.insert(1)
t.insert(6)
t.insert(4)
t.insert(7)
t.insert(10)
t.insert(14)
t.insert(13)
t.preShow(t.getRoot())