leetcode.203
题目要求
单链表的定义
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; }
}
不添加虚拟头节点
特殊情况
1.链表全空
2.head.val == val
代码
public ListNode removeElements(ListNode head, int val) {
if(head == null) return head;//全空,直接返回
ListNode temp = head;
while(temp.next != null) {
if(temp.next.val == val) {
temp.next = temp.next.next;
} else temp = temp.next;
}
if(head.val == val) head = head.next;
return head;
}
添加虚拟头节点
如果添加一个虚拟头节点,next指向head,那么以上的两个情况则不需要特殊考虑。
代码
public ListNode removeElements(ListNode head, int val) {
ListNode newHead = new ListNode(-1, head);
ListNode temp = newHead;
while(temp.next != null) {
if(temp.next.val == val) {
temp.next = temp.next.next;
} else temp = temp.next;
}
return newHead.next;
}