Update insert

mgechev · mgechev · commit 4466fb4c7b40 · 2014-05-16T16:18:37.000-07:00
diff --git a/src/data-structures/red-black-tree.js b/src/data-structures/red-black-tree.js
@@ -42,21 +42,22 @@
   }
 
   RBTree.prototype.put = function (key, value) {
-    return (this._root = this._put(key, value, this._root));
+    this._root = this._put(key, value, this._root);
+    this._root.flipColor();
   };
 
   RBTree.prototype._put = function (key, value, node) {
     var newRoot = node;
     if (this._root === null) {
-      return new Node(key, value, null, null, false);
+      return new Node(key, value, null, null, true);
     }
-    if (node.getValue() > value) {
+    if (node.getKey() > key) {
       this._put(key, value, node.getLeft());
-    } else if (node.getValue() < value) {
-      this._put(key, value, node.getRight());
+    } else if (node.getKey() < key) {
+      node.setLeft(this._put(key, value, node.getRight()));
     }
     if (this._isRed(node.getRight())) {
-      newRoot = this._rotateLeft(node);
+      node.setRight(this._rotateLeft(node));
     }
     if (this._isRed(node.getLeft()) && this._isRed(node.getLeft().getLeft())) {
       newRoot = this._rotateRight(node);
diff --git a/test/data-structures/red-black-tree.spec.js b/test/data-structures/red-black-tree.spec.js
@@ -53,5 +53,13 @@ describe('RBTree', function () {
       expect(tree._root.getKey()).toBe('foo');
       expect(tree._root.getValue()).toBe('bar');
     });
+
+    it('should be able to insert a node in 1 level tree', function () {
+      var tree = new RBTree();
+      tree.put(1, 'bar');
+    
+    });
+
   });
+
 });
