Added 'remove' method to remove by element in CustomLinkedList

zevolution · zevolution · commit bf457b387013 · 2021-03-15T23:16:35.000-03:00
diff --git a/src/main/java/br/com/zevolution/datastructure/linkedlist/CustomLinkedList.java b/src/main/java/br/com/zevolution/datastructure/linkedlist/CustomLinkedList.java
@@ -58,6 +58,33 @@ public void removeFirst() {
 		}
 	}
 
+	public void remove(Object element) {
+		Node current = this.first;
+		Node previous = null;
+
+		if (current.getElement() == element) {
+			this.first = current.getNext();
+			current = null;
+			this.totalElements--;
+			return;
+		}
+
+		while (current != null) {
+			if (current.getElement() == element) {
+				break;
+			}
+
+			previous = current;
+			current = current.getNext();
+		}
+
+		if (current == null) return;
+
+		previous.setNext(current.getNext());
+		current = null;
+		this.totalElements--;
+	}
+
 	public Object get(int position) {
 		return this.getNode(position).getElement();
 	}
diff --git a/src/test/java/br/com/zevolution/datastructure/linkedlist/CustomLinkedListTest.java b/src/test/java/br/com/zevolution/datastructure/linkedlist/CustomLinkedListTest.java
@@ -93,4 +93,18 @@ public void should_ThrowException_When_NoSuchElement() {
 		linkedList.removeFirst();
 	}	
 
+	@Test
+	public void should_RemoveMiddleElement() {
+		CustomLinkedList linkedList = new CustomLinkedList();
+		linkedList.add("Bia");
+		linkedList.add("Lucas");
+		linkedList.add("Laura");
+
+		linkedList.remove("Lucas");
+
+		assertEquals(2, linkedList.size());
+		assertEquals("Bia", linkedList.get(0));
+		assertEquals("Laura", linkedList.get(1));
+	}
+
 }
