-
Notifications
You must be signed in to change notification settings - Fork 0
/
btusingpython.py
72 lines (65 loc) · 2.22 KB
/
btusingpython.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
#lesson no : 22 binary tree
#binary tree using python list
class BT:
def __init__(self,size):
self.customlist=size*[None]
self.lastusedindex=0
self.maxsize=size
def insertnode(self,value):
if self.lastusedindex+1==self.maxsize:
return "BT is full"
self.customlist[self.lastusedindex+1]=str(value)
self.lastusedindex+=1
return 'node has been sucessfully inserted'
def searchnode(self,nodevalue):
for i in range(len(self.customlist)):
if self.customlist[i]==nodevalue:
return self.customlist[i]
else:
return 'not found'
def preordeordertraversal(self,index):
if index<self.lastusedindex:
return
else:
print(self.customlist[index])
self.preordeordertraversal(index * 2)
self.preordeordertraversal(index * 2 + 1)
def inordertraversal(self,index):
if index<self.lastusedindex:
return
else:
self.preordeordertraversal(index * 2)
print(self.customlist[index])
self.preordeordertraversal(index * 2 + 1)
def postordertraversal(self,index):
if index<self.lastusedindex:
return
else:
self.preordeordertraversal(index * 2)
self.preordeordertraversal(index * 2 + 1)
print(self.customlist[index])
def levelordertraversal(self,index):
for i in range(index,self.lastusedindex+1):
print(self.customlist[i])
def deletenode(self,value):
if self.lastusedindex==0:
return 'BT does not exist'
for i in range(1,self.lastusedindex+1):
if self.customlist[i]==value:
self.customlist[i]=self.customlist[self.lastusedindex]
self.lastusedindex=None
self.lastusedindex+=1
return 'node successfully deleted'
def delete(self):
self.customlist=None
return
nodebt=BT(10)
nodebt.insertnode('Drinks')
nodebt.insertnode('hot')
nodebt.insertnode('cold')
nodebt.insertnode('tea')
nodebt.insertnode('cofee')
nodebt.insertnode('alcholic')
nodebt.insertnode('non alcholic')
nodebt.delete()
nodebt.levelordertraversal(1)