1.题目描述:
给你一个链表的头节点head和一个整数val,请你删除链表中所有满足Node.val == val的节点,并返回新的头节点。
2.构建虚拟头节点:
涉及到头节点删除操作逻辑会不同,一般删除的逻辑为temp.next = temp.next.next,根据temp.next的val值来找到待删除元素,temp.next为末位节点也无妨。根据构建一个虚拟头节点,即可全部采用相同的判断逻辑。
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode() {}
* ListNode(int val) { this.val = val; }
* ListNode(int val, ListNode next) { this.val = val; this.next = next; }
* }
*/
class Solution {
public ListNode removeElements(ListNode head, int val) {
if (head == null) return head;
ListNode virHead = new ListNode();
virHead.next = head;
ListNode temp = virHead;
while (temp.next != null) {
if (temp.next.val == val) {
temp.next = temp.next.next;//删除操作,这里不能temp后移,否则无法判断temp.next节点是否删除
} else {
temp = temp.next;//无删除操作可将temp后移
}
}
return virHead.next;
}
}
3.不构建虚拟头节点:
就是单独把头节点删除的逻辑抽取出来,其他代码和上面几乎一样。
class Solution {
public ListNode removeElements(ListNode head, int val) {
while (head != null && head.val == val) {
head = head.next;
}
if (head == null) return head;
ListNode temp = head;//走到这里head必不为val
while (temp.next != null) {
if (temp.next.val == val) {
temp.next = temp.next.next;
} else {
temp = temp.next;
}
}
return head;
}
}