Update Checker.py

Algorithms_and_Data_Structures/TimeComplexity_analyzer/Checker.py

@@ -3,21 +3,29 @@
 class TimeComplexityAnalyzer(ast.NodeVisitor):
     def __init__(self):
         self.complexity = 0
+        self.loop_depth = 0
 
     def visit_For(self, node):
-        self.complexity += 1
+        self.loop_depth += 1
+        self.complexity += 2 ** self.loop_depth
         self.generic_visit(node)
+        self.loop_depth -= 1
 
     def visit_While(self, node):
-        self.complexity += 1
+        self.loop_depth += 1
+        self.complexity += 2 ** self.loop_depth
         self.generic_visit(node)
+        self.loop_depth -= 1
 
     def visit_FunctionDef(self, node):
         self.generic_visit(node)
 
     def visit_Call(self, node):
         self.generic_visit(node)
 
+    def visit_If(self, node):
+        self.generic_visit(node)
+
     def get_complexity(self):
         if self.complexity == 0:
             return "O(1) - Constant Time"