63 lines
1.3 KiB
JavaScript
63 lines
1.3 KiB
JavaScript
// https://www.freecodecamp.org/learn/coding-interview-prep/data-structures/remove-elements-from-a-linked-list-by-index
|
|
|
|
function LinkedList() {
|
|
var length = 0;
|
|
var head = null;
|
|
|
|
var Node = function (element) {
|
|
this.element = element;
|
|
this.next = null;
|
|
};
|
|
|
|
this.size = function () {
|
|
return length;
|
|
};
|
|
|
|
this.head = function () {
|
|
return head;
|
|
};
|
|
|
|
this.add = function (element) {
|
|
var node = new Node(element);
|
|
if (head === null) {
|
|
head = node;
|
|
} else {
|
|
var currentNode = head;
|
|
|
|
while (currentNode.next) {
|
|
currentNode = currentNode.next;
|
|
}
|
|
|
|
currentNode.next = node;
|
|
}
|
|
|
|
length++;
|
|
};
|
|
|
|
// Only change code below this line
|
|
this.removeAt = (index) => {
|
|
if (index < 0 || index >= length || head === null) {
|
|
return null;
|
|
}
|
|
let i = 0;
|
|
let previousNode = null;
|
|
let currentNode = head;
|
|
while (i < index && currentNode !== null) {
|
|
i++;
|
|
previousNode = currentNode;
|
|
currentNode = currentNode.next;
|
|
}
|
|
if (currentNode === null) {
|
|
return null;
|
|
}
|
|
if (previousNode === null) {
|
|
head = currentNode.next;
|
|
} else {
|
|
previousNode.next = currentNode.next;
|
|
}
|
|
length--;
|
|
return currentNode.element;
|
|
};
|
|
// Only change code above this line
|
|
}
|