feat: add checkIsBalanced method to BST

Author: soloVlad

binary_search_tree/Tree.js

@@ -161,6 +161,23 @@ class Tree {
     }
   }
 
+  checkIsBalanced(currentNode = this.root) {
+    if (!currentNode) {
+      return true;
+    }
+
+    const leftHeight = this.height(currentNode.left);
+    const rightHeight = this.height(currentNode.right);
+
+    const difference = Math.abs(leftHeight - rightHeight);
+
+    return (
+      difference <= 1
+      && this.checkIsBalanced(currentNode.left)
+      && this.checkIsBalanced(currentNode.right)
+    );
+  }
+
   print(node = this.root, prefix = "", isLeft = true) {
     if (node === null) {
       return;

binary_search_tree/index.js

@@ -26,4 +26,11 @@ console.log(tree.inOrder());
 const seven = tree.find(7);
 
 console.log(tree.height(seven));
-console.log(tree.depth(seven));
+console.log(tree.depth(seven));
+
+console.log(tree.checkIsBalanced());
+
+tree.insert(2.5);
+tree.print();
+
+console.log(tree.checkIsBalanced());