Move tree height to a class

2026-01-07 18:08:01 +00:00 · 2016-08-14 08:29:25 -04:00
parent b4ed90e649
commit 9176ffe214
4 changed files with 42 additions and 56 deletions
--- a/graphs_trees/tree_height/height.py
+++ b/graphs_trees/tree_height/height.py
@@ -1,5 +1,7 @@
-def height(node):
-    if node is None:
-        return 0
-    return 1 + max(height(node.left), 
-                   height(node.right))
+class BstHeight(Bst):
+
+    def height(self, node):
+        if node is None:
+            return 0
+        return 1 + max(self.height(node.left),
+                       self.height(node.right))