Fixed assumption of root starting with 0 (#259)

codezonediitj · Apr 19, 2020 · 8d8e688 · 8d8e688
1 parent 4452dc8
commit 8d8e688
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pydatastructs/trees/binary_trees.py b/pydatastructs/trees/binary_trees.py
@@ -218,7 +218,7 @@ def insert(self, key, data=None):
             self.tree[walk].key = key
             self.tree[walk].data = data
             return None
-        new_node, prev_node, flag = TreeNode(key, data), 0, True
+        new_node, prev_node, flag = TreeNode(key, data), self.root_idx, True
         while flag:
             if not self.comparator(key, self.tree[walk].key):
                 if self.tree[walk].right is None: