Skip to content

Commit

Permalink
Add Binary Search Tree in data_structures
Browse files Browse the repository at this point in the history
  • Loading branch information
frmatias committed Jun 20, 2017
1 parent 65de84b commit 4a9405e
Show file tree
Hide file tree
Showing 2 changed files with 102 additions and 0 deletions.
1 change: 1 addition & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -88,3 +88,4 @@ ENV/
# Rope project settings
.ropeproject
.idea
.DS_Store
101 changes: 101 additions & 0 deletions data_structures/Binary Tree/binary_seach_tree.py
Original file line number Diff line number Diff line change
@@ -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())

0 comments on commit 4a9405e

Please sign in to comment.