Update BinarySearchTree.js

JaapvanEkris · Nov 28, 2023 · b64d7b7 · b64d7b7
1 parent 33923db
commit b64d7b7
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/app/engine/utils/BinarySearchTree.js b/app/engine/utils/BinarySearchTree.js
@@ -12,6 +12,24 @@ function createLabelledBinarySearchTree () {
   function push (label, value) {
     if (tree === null) {
       tree = newNode(label, value)
+    } else {
+      pushInTree(tree, label, value)
+    }
+  }
+
+  function pushInTree (currentTree, label, value) {
+    if (currentTree === null) {
+      currentTree = newNode(label, value)
+    } else {
+      // We encounter a filled node
+      if (currentTree.value >= value) {
+        // value <= currentTree.value, so we need the value to the left branch
+        pushInTree (currentTree.leftNode, label, value)
+      } else {
+        // currentTree.value < value, so we need to add the value to the right branch
+        pushInTree (currentTree.rightNode, label, value)
+      }
+      currentTree.numberOfLeafsAndNodes = currentTree.numberOfLeafsAndNodes + 1
     }
   }