diff --git a/src/main/java/com/williamfiset/algorithms/datastructures/binarysearchtree/BinarySearchTree.java b/src/main/java/com/williamfiset/algorithms/datastructures/binarysearchtree/BinarySearchTree.java index d48e7f1a6..6fa356059 100644 --- a/src/main/java/com/williamfiset/algorithms/datastructures/binarysearchtree/BinarySearchTree.java +++ b/src/main/java/com/williamfiset/algorithms/datastructures/binarysearchtree/BinarySearchTree.java @@ -51,7 +51,9 @@ public boolean add(T elem) { // Otherwise add this element to the binary tree } else { - root = add(root, elem); + node = add(root, elem); + if (isEmpty()) + root = node nodeCount++; return true; }