Made more options

Author: assaabriiii

Node.py

@@ -229,6 +229,21 @@ def search(self, node: Node , Arg: int) :
         elif node.value < Arg:
             return self.search(node.right, Arg)
 
-        
-B = BinaryTree([10,1,12])
-print(B.search(B.root,15))
+    
+    def FullTree(node: Node) :
+        queue = []
+        queue.append(node)
+
+        while queue != []:
+            temp = queue.pop()
+
+            if temp.left == None and temp.right == None:
+                continue
+            elif temp.right != None and temp.left != None:
+                queue.append(temp.right)
+                queue.append(temp.left)
+            else:
+                return False
+        return True
+        
+        

__pycache__/Node.cpython-310.pyc

@@ -56,9 +56,9 @@ def draw(node, x, y, dx):
     t.hideturtle()
     turtle.mainloop()
 
-if __name__ == '__main__':
-    Banching([2,3,1,4,5])
-    drawtree()
+# if __name__ == '__main__':
+#     Banching([2,3,1,4,5])
+#     drawtree()
 
 
 

__pycache__/Node.cpython-311.pyc

@@ -2,6 +2,7 @@
 from os import system
 from Node import * 
 from time import sleep 
+import random 
 
 def display_menu(menu):
     """
@@ -12,9 +13,15 @@ def display_menu(menu):
     print(Fore.LIGHTBLUE_EX + "  -> " + Fore.WHITE + " Binary Tree Options " + Fore.LIGHTBLUE_EX + " <-\n\n")
 
     try : 
-        print("Current tree -> " , tree.data ,"\n\n")
+        print("Main tree -> " , tree.data ,"\n\n")
     except :
         print("Tree is not defined\n\n")
+    
+    try : 
+         print("Second tree -> " , lary.data ,"\n\n")
+    except :
+        print("Second tree is not defined\n\n")
+        
 
     for k, function in menu.items():
         print(Fore.MAGENTA ,"|",k,"| -> ", Fore.YELLOW ,function.__name__)
@@ -25,22 +32,65 @@ def Binarytree():
     print("you have selected menu option BinaryTree making") # Simulate function output.
     Nodes = []
     global tree
+    select = input("Manually(1) or random(2) ? ")
+    if select == "1" :
+        system("clear")
+        while True : 
+            n = int(input("Enter the node : "))
+            system("clear")
+            if (n != -1) : 
+                Nodes.append(n)
+        
+            else : 
+                Tree = BinaryTree(Nodes)
+                tree = Tree 
+                break
+    
+    elif select == "2" : 
+        system("clear")
+        rang = int(input("How many nodes do you want to be generated ? "))
+        
+        for i in range(rang):
+            r = random.randint(0,100)
+            if r in Nodes : 
+                continue 
+            else : 
+                Nodes.append(r)
+        Tree = BinaryTree(Nodes)
+        tree = Tree 
+        
+    
+    print(Tree.data)
+    input("Press Enter to Continue\n")
+    system('clear')  # clears stdout
+
+def Comparing() : 
+    system("clear")
+    print("you have selected menu option Comparing") # Simulate function output.
+    Nodes = []
+    global lary 
     while True : 
         n = int(input("Enter the node : "))
         system("clear")
         if (n != -1) : 
             Nodes.append(n)
 
         else : 
-            Tree = BinaryTree(Nodes)
-            tree = Tree 
+            Lary = BinaryTree(Nodes)
+            lary = Lary 
             break
+    print(Lary.data)
+    input("Press Enter to compare\n")
+    
+    if lary.PrintTree(str(1)) == tree.PrintTree(str(1)) : 
+        print("True")
+    else : 
+        print("False")
 
-    print(Tree.data)
     input("Press Enter to Continue\n")
     system('clear')  # clears stdout
 
-
+        
 def Draw():
     system("clear")
     print("you have selected menu option drawing") # Simulate function output.
@@ -117,7 +167,21 @@ def Search() :
     input("Press Enter to Continue\n")
     system('clear')  # clears stdout
 
-
+def CheckForFullTree() : 
+    system("clear")
+    print("you have selected menu option Checking for full tree ") # Simulate function output.
+    print(tree.FullTree())
+    
+    input("Press Enter to Continue\n")
+    system('clear')  # clears stdout
+    
+def Contributors() : 
+    system("clear")
+    print(Fore.RESET,"\nAmirhossein Sabry 40011573\nKimia Keivanloo 40015753\n\nWWW.GEEKFORGEEKS.COM \u2764\uFE0F")
+    sleep(5)
+    input("Press Enter to Continue\n")
+    system('clear')  # clears stdout
+    
 def done():
     system('clear')  # clears stdout
     print("Goodbye")
@@ -130,7 +194,7 @@ def get_word():
 def main():
     # Create a menu dictionary where the key is an integer number and the
     # value is a function name.
-    functions_names = [Binarytree, Draw, MAX_and_MIN, CountLeafs , DeleteTree , numberOfFloors ,traversal, Search , done]
+    functions_names = [Binarytree,Comparing, Draw, MAX_and_MIN, CountLeafs , DeleteTree , numberOfFloors ,traversal, Search ,CheckForFullTree, Contributors ,done]
     menu_items = dict(enumerate(functions_names, start=1))
 
     while True:
@@ -144,4 +208,5 @@ def main():
 if __name__ == "__main__":
     from colorama import * 
     print("\n",Fore.WHITE , Back.RED ,"Please define the tree before using other options !!!",Back.RESET , "\n")
+    
     main()