File tree Expand file tree Collapse file tree 1 file changed +10
-3
lines changed Expand file tree Collapse file tree 1 file changed +10
-3
lines changed Original file line number Diff line number Diff line change
1
+ # Class to build a tree node
1
2
class TreeNode :
3
+ # Constructor to create tree node
2
4
def __init__ (self , data ):
3
5
self .data = data
4
6
self .children = []
5
7
self .parent = None
6
-
8
+
9
+ # Function to add a add in individual node
7
10
def add_child (self , child ):
8
11
child .parent = self
9
12
self .children .append (child )
10
-
13
+
14
+ # Function to get a level of the selected node
11
15
def getlevel (self ):
12
16
level = 0
13
17
p = self .parent
14
18
while p :
15
19
level += 1
16
20
p = p .parent
17
21
return level
18
-
22
+
23
+ # Print the contructed tree
19
24
def printt (self ):
20
25
prefix = (" " * 4 * self .getlevel ()) + ("|--" if self .parent else "" )
21
26
print (prefix + self .data )
@@ -24,7 +29,9 @@ def printt(self):
24
29
child .printt ()
25
30
26
31
32
+ # Funcation to build a tree
27
33
def build_tree ():
34
+ # Taking input dynamically
28
35
print ("Do the following given in eg bracket" )
29
36
print ("\n " )
30
37
root = input ('Enter a root node element (eg: Food): ' )
You can’t perform that action at this time.
0 commit comments