leetcode 203. 移除链表元素

给你一个链表的头节点 head 和一个整数 val ，请你删除链表中所有满足 Node.val == val 的节点，并返回 新的头节点 。

示例 1：

输入：head = [1,2,6,3,4,5,6], val = 6
输出：[1,2,3,4,5]

示例 2：

输入：head = [], val = 1
输出：[]

示例 3：

输入：head = [7,7,7,7], val = 7
输出：[]

提示：

列表中的节点数目在范围 [0, 104] 内
1 <= Node.val <= 50
0 <= val <= 50

思路分析

分两种情况判断

1.当head不为空，并等于val时，让下一个节点覆盖掉头节点
2.中间节点等于val的情况，让cur指向head，避免prev使代码更简洁。
3.循环判断，返回头节点

不设置虚拟节点

class Solution {
    public ListNode removeElements(ListNode head, int val) {
        while(head != null && head.val == val){//头节点等于val的一种情况
            head = head.next;//覆盖头节点
        }
        //删除头节点不等于val的元素情况
        ListNode cur = head; 
        while(cur != null){
            while(cur.next != null && cur.next.val == val){
                cur.next = cur.next.next;
            }
            cur = cur.next;//如果不等于val向下走一位
        } 
        return head;
    }
}

添加虚拟头节点

class Solution {
    public ListNode removeElements(ListNode head, int val) {
        ListNode dummyHead = new ListNode(0);
        dummyHead.next = head;
        ListNode cur = dummyHead;
        while(cur.next != null){
            if(cur.next.val == val){
                cur.next = cur.next.next;
            }else{
                //val在中间
                cur = cur.next;
            }
        }
        return dummyHead.next;
    }
}