Update treenode.py

Author: ivanmmarkovic

Ch06. Trees and Tree Algorithms/bst/treenode.py

@@ -1,51 +1,38 @@
 
+
 class TreeNode:
     def __init__(self, key = None, parent = None, leftChild = None, rightChild = None):
         self.key = key
         self.parent = parent
         self.leftChild = leftChild
         self.rightChild = rightChild
 
-    def isRoot(self)->bool:
-        return self.parent is None
-
-    def hasParent(self)->bool:
-        return self.parent is not None
-
-    def isLeftChild(self)->bool:
-        return self.parent.leftChild == self
-
-    def isRightChild(self)->bool:
-        return self.parent.rightChild == self
-
     def hasLeftChild(self)->bool:
         return self.leftChild is not None
 
     def hasRightChild(self)->bool:
         return self.rightChild is not None
 
-    def hasBothChildren(self):
+    def hasBothChildren(self)->bool:
         return self.hasLeftChild() and self.hasRightChild()
 
-    def isLeaf(self):
+    def isLeaf(self)->bool:
         return not self.hasLeftChild() and not self.hasRightChild()
 
+    def isLeftChild(self)->bool:
+        return self.parent.leftChild == self
+
+    def isRightChild(self)->bool:
+        return self.parent.rightChild == self
+
     def findMin(self):
-        if self.hasLeftChild():
-            return self.leftChild.findMin()
-        else:
-            return self
-        
+        node = self
+        while node.leftChild is not None:
+            node = node.leftChild
+        return node
+
     def findMax(self):
-        if self.hasRightChild():
-            return self.rightChild.findMax()
-        else:
-            return self
-
-    def inorder(self):
-        if self is not None:
-            if self.leftChild is not None:
-                self.leftChild.inorder()
-            print(self.key, end=", ")
-            if self.rightChild is not None:
-                self.rightChild.inorder()
+        node = self
+        while node.rightChild is not None:
+            node = node.rightChild
+        return node