added 2 methods to Node (tree.py): depth_down() (name could be bettern, though) and width()

1 files changed, 14 insertions, 0 deletions

diff --git a/tree.py b/tree.py
index 87f7076..551c2d2 100644
--- a/tree.py
+++ b/tree.py
@@ -163,6 +163,20 @@ class Node :
         else :
             return 0
 
+    def depth_down(self):
+        """
+        return depth of the tree from this node
+        """
+        if self.children:
+            return 1 + max([c.depth_down() for c in self.children])
+        return 1
+
+    def width(self):
+        """
+        return the width of the tree from this node
+        """
+        return len(self.leaves())
+        
     def root(self):
         """
         return the root node of the tree