Made size a method of nodes too

6 years ago · d904f5e1ca
1 changed files with 9 additions and 0 deletions
--- a/BST.py
+++ b/BST.py
@ -4,6 +4,15 @@ class BSTNode:
        self.left = None
        self.right = None

+    def size(self):
+        ls = 0
+        rs = 0
+        if self.left is not None:
+            ls = self.left.size()
+        if self.right is not None:
+            rs = self.right.size()
+        return rs + ls + 1
+
    def height(self):
        lh = 0
        rh = 0