第二章 链表_203.移除链表元素

203.移除链表元素

一、题目

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

二、思考

1. 没有设置虚拟节点

没有设置虚拟节点：
如果头结点是指定的值，就一直循环头结点；如果头结点最终为空，就返回null
如果头结点的下一个节点为空，直接返回头结点即可
preNode=head，curNode=head.next
循环判断curNode
if (curNode.val == val) {
preNode.next = curNode.next;
} else {
preNode = curNode;
}
curNode = curNode.next;
如果没有设置虚拟节点，最终返回的是 head

2. 设置类虚拟节点

通过虚拟节点来返回最终结果，虚拟节点不能移动，作为 head 节点的前节点
非空判断
单独定义虚拟节点 dummy，作为链表的第一个节点
设定 preNode 节点；设置 curNode 节点
循环判断 curNode
if (curNode.val == val) {
preNode.next = curNode.next;
} else {
preNode = curNode;
}
curNode = curNode.next;
由于设置了虚拟节点，所以最终的返回结果是 dummy.next

三、遇到的问题

1. 由于这里的头结点设置的是 preNode，但是 preNode 一直在移动，所以最后就无法根据 preNode.next 来获取下一个节点

解决办法：单独设置一个节点 用来指向头结点，并且不移动这个节点

四、代码

1. 无虚拟头结点

public ListNode removeElements1(ListNode head, int val) {
        if (head == null) {
            return null;
        }
        while (head != null && head.val == val) {
            head = head.next;
        }
        if (head == null) {
            return null;
        }
        ListNode preNode = head;
        ListNode curNode = null;
        if (head.next != null) {
            curNode = head.next;
        }
        while (curNode != null) {
            if (curNode.val == val) {
                preNode.next = curNode.next;
            } else {
                preNode = curNode;
            }
            curNode = curNode.next;
        }
        return head;
    }

2. 有虚拟头结点

public ListNode removeElements2(ListNode head, int val) {
        if (head == null) {
            return null;
        }
        ListNode dummy = new ListNode(-1, head);
        ListNode preNode = dummy;
        ListNode curNode = head;
        while (curNode != null) {
            if (curNode.val == val) {
                preNode.next = curNode.next;
            } else {
                preNode = curNode;
            }
            curNode = curNode.next;
        }
        return dummy.next;
    }