From d904f5e1cab954db0b60ea2e8dbde33400ae1526 Mon Sep 17 00:00:00 2001 From: Raphael Roberts Date: Wed, 13 Nov 2019 16:29:49 -0600 Subject: [PATCH] Made size a method of nodes too --- BST.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/BST.py b/BST.py index ddbe41f..7dbdff4 100644 --- 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