-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathbinary_tree.py
More file actions
33 lines (33 loc) · 1014 Bytes
/
Copy pathbinary_tree.py
File metadata and controls
33 lines (33 loc) · 1014 Bytes
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
class BinaryTree:
def __init__(self, root):
self.root = root
self.left_node = None
self.right_node = None
def insert_left_node(self, data):
if self.left_node is None:
self.left_node = BinaryTree(data)
else:
new_node = BinaryTree(data)
new_node.left_node = self.left_node
self.left_node = new_node
def insert_right_node(self, data):
if self.right_node is None:
self.right_node = BinaryTree(data)
else:
new_node = BinaryTree(data)
new_node.right_node = self.right_node
self.right_node = new_node
def get_right_node(self):
return self.right_node
def get_left_node(self):
return self.left_node
def set_root(self, root):
self.root = root
def get_root(self):
return self.root
b = BinaryTree(0)
b.insert_left_node(1)
b.insert_right_node(10)
print(b.left_node.root)
print(b.right_node.root)
print(b.get_root())