移除单链表元素(Java)

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;
    }