题目:
Remove all elements from a linked list of integers that have value val.
Example
Given: 1 --> 2 --> 6 --> 3 --> 4 --> 5 --> 6, val = 6
Return: 1 --> 2 --> 3 --> 4 --> 5
本题是删除链表中指定值的元素,在这里总结一下链表删除操作的方法:
- 第一种,可以选择删除当前元素的下一个元素,代码很简单,
tmp.next = tmp.next.next
- 第二种,可以删除当前元素,即将当前元素变为下一个元素,代码入下:
tmp.val = tmp.next.val; tmp.next = tmp.next.next
针对本体而言,先使用第一种方法,但是问题在于,如果链表的第一个元素就是要删除的元素应该如何应对呢,这里提供两种方法,一种是先判断,如果是就一直删除直到不是,另一种是先不管第一个元素,当把后面的搞完之后在判断。代码如下所示:
public ListNode removeElements1(ListNode head, int val) {
//先将前面满足删除条件的元素全部删除掉
while (head != null && head.val == val) head = head.next;
ListNode curr = head;
while (curr != null && curr.next != null)
if (curr.next.val == val) curr.next = curr.next.next;
else curr = curr.next;
return head;
}
public ListNode removeElements3(ListNode head, int val) {
if (head == null) return null;
ListNode pointer = head;
while (pointer.next != null) {
if (pointer.next.val == val) pointer.next = pointer.next.next;
else pointer = pointer.next;
}
//直到最后在判断第一个元素是否需要删除
return head.val == val ? head.next : head;
}
此外,我们还可以使用递归函数的方法来实现。代码入下:
public ListNode removeElements2(ListNode head, int val) {
if (head == null) return null;
head.next = removeElements2(head.next, val);
return head.val == val ? head.next : head;
}