From 5a0fee02ddca624e0d99d02381f3e5f6d89f3ce0 Mon Sep 17 00:00:00 2001 From: Mitanshu Holkar Date: Fri, 21 Oct 2022 10:36:09 +0530 Subject: [PATCH] Create binarytree.py --- binarytree.py | 36 ++++++++++++++++++++++++++++++++++++ 1 file changed, 36 insertions(+) create mode 100644 binarytree.py diff --git a/binarytree.py b/binarytree.py new file mode 100644 index 0000000..40f815f --- /dev/null +++ b/binarytree.py @@ -0,0 +1,36 @@ +class Node: + def __init__(self, data): + self.left = None + self.right = None + self.data = data + + def insert(self, data): +# Compare the new value with the parent node + if self.data: + if data < self.data: + if self.left is None: + self.left = Node(data) + else: + self.left.insert(data) + elif data > self.data: + if self.right is None: + self.right = Node(data) + else: + self.right.insert(data) + else: + self.data = data + +# Print the tree + def PrintTree(self): + if self.left: + self.left.PrintTree() + print( self.data), + if self.right: + self.right.PrintTree() + +# Use the insert method to add nodes +root = Node(12) +root.insert(6) +root.insert(14) +root.insert(3) +root.PrintTree()