203. 移除链表元素https://leetcode.cn/problems/remove-linked-list-elements/

1、题目

给你一个链表的头节点 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

2、题解

题解1

头结点单独处理。

/**
 * 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 {
    // 移除链表中所有值等于val的节点
    public ListNode removeElements(ListNode head, int val) {
        
        // 1. 移除头节点的值等于val的情况
        // 如果当前头节点的值等于val，继续移动头节点，直到头节点值不等于val
        while(head != null && head.val == val) {
            head = head.next;  // 将头节点更新为下一个节点
        }

        // 2. 定义指针p从新的头节点开始遍历
        ListNode p = head;
        
        // 3. 遍历链表，删除所有值等于val的节点
        // 判断当前节点的下一个节点是否需要删除
        while(p != null && p.next != null) {
            // 如果下一个节点的值等于val，跳过该节点
            if(p.next.val == val) {
                p.next = p.next.next;  // 跳过下一个节点
            } else {
                p = p.next;  // 否则继续移动指针
            }
        }

        // 4. 返回修改后的链表头
        return head;
    }
}

题解2

使用一个虚拟头结点。

/**
 * 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 {
    // 移除链表中所有值等于val的节点
    public ListNode removeElements(ListNode head, int val) {
        
        // 1. 创建一个虚拟头节点 prehead，指向链表的头部
        // 这样可以方便处理头节点被删除的情况
        ListNode prehead = new ListNode(-1, head);
        
        // 2. 定义一个指针p从prehead开始遍历链表
        // prehead的next指向head，p会遍历整个链表
        ListNode p = prehead;
        
        // 3. 遍历链表
        // 当p的下一个节点不为null时继续循环
        while(p != null && p.next != null) {
            
            // 4. 判断当前节点的下一个节点是否需要删除
            // 如果下一个节点的值等于val，则跳过该节点
            if(p.next.val == val) {
                p.next = p.next.next;  // 删除当前节点的下一个节点
            } else {
                p = p.next;  // 否则，继续移动p指针到下一个节点
            }
        }
        
        // 5. 返回修改后的链表头部
        // prehead.next就是去除指定值后的链表的头节点
        return prehead.next;
    }
}