移除链表中的元素Day-10

移除链表中的元素

给你一个链表的头节点 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
输出：[]

题解

/**
 * 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) {
        //定义虚拟头节点，赋值虚拟头节点为 0
        ListNode dummy = new ListNode(0);
        //定义前驱节点， 指向虚拟节点
        ListNode pre = dummy;
        //当前节点指向头节点
        ListNode cur = head;
        //虚拟头节点 连接 头节点
        dummy.next = head;

        while(cur != null){
            //如果找到就pre的下一个指向当前节点的下一个（删除cur元素）
            if(cur.val == val){
                pre.next = cur.next;
            }else{
                //如果没有，向前走
                pre = cur;
            }
            //向下走
            cur = cur.next;
        }
        //返回虚拟头节点指向的下一个节点
        return dummy.next;
    }
}