Merge pull request amejiarosario#15 from amejiarosario/fix-linked-list-remove-first

fix(linkedlist): removeFirst's last reference

Author: amejiarosario

src/data-structures/linked-lists/linked-list.js

@@ -183,10 +183,10 @@ class LinkedList {
       this.first = head.next;
       if (this.first) {
         this.first.previous = null;
+      } else {
+        this.last = null;
       }
       this.size -= 1;
-    } else {
-      this.last = null;
     }
     return head && head.value;
   }

src/data-structures/linked-lists/linked-list.spec.js

@@ -32,6 +32,17 @@ describe('LinkedList Test', () => {
       linkedList.addLast('b');
     });
 
+    it('should remove with only one element', () => {
+      linkedList = new LinkedList();
+      linkedList.addLast('a');
+
+      expect(linkedList.removeFirst()).toBe('a');
+
+      expect(linkedList.size).toBe(0);
+      expect(linkedList.first).toBe(null);
+      expect(linkedList.last).toBe(null);
+    });
+
     it('should remove first item: a', () => {
       expect(linkedList.removeFirst()).toBe('a');
       expect(linkedList.first.value).toBe('b');
@@ -44,6 +55,7 @@ describe('LinkedList Test', () => {
 
       expect(linkedList.first).toBe(null);
       expect(linkedList.last).toBe(null);
+      expect(linkedList.size).toBe(0);
     });
   });