哨兵节点——标准工具人

原题

这还是一道链表题，原题如下：

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

解题思路

常规解法

如果删除的节点是中间的节点，则问题似乎非常简单：

• 选择要删除节点的前一个结点 prev。
• 将 prev 的 next 设置为要删除结点的 next。

当要删除的一个或多个节点位于链表的头部时，事情会变得复杂。

首先要用一个while循环对头节点进行判断，删除所有头部要删除的节点，在按上述方法删除中间的节点。具体代码如下：

public static ListNode myRemoveElements(ListNode head, int val) {
    // 删除头节点
    while (head != null) {
        if (head.val == val) {
            head = head.next;
        } else {
            break;
        }
    }
    // 正常判断next
    ListNode current = head;
    while (current != null && current.next != null) {
        if (current.next.val == val) {
            current.next = current.next.next;
        } else {
            current = current.next;
        }
    }
    return head;
}

哨兵节点解法

引入哨兵节点，无需再分步判断头节点和中间节点了。

哨兵节点广泛应用于树和链表中，如伪头、伪尾、标记等，它们是纯功能的，通常不保存任何数据，其主要目的是使链表标准化，如使链表永不为空、永不无头、简化插入和删除。即标准的工具人。

在这里哨兵节点将被用于伪头。

算法：

• 初始化哨兵节点为 ListNode(0) 且设置 sentinel.next = head。
• 初始化两个指针 curr 和 prev 指向当前节点和前继节点。
• 当 curr != nullptr：
  • 比较当前节点和要删除的节点：
    • 若当前节点就是要删除的节点：则 prev.next = curr.next。
    • 否则设 prve = curr。
  • 遍历下一个元素：curr = curr.next。
• 返回 sentinel.next。

public static ListNode removeElements(ListNode head, int val) {
    ListNode sentinel = new ListNode(0);
    sentinel.next = head;

    ListNode prev = sentinel, curr = head;
    while (curr != null) {
        if (curr.val == val) prev.next = curr.next;
        else prev = curr;
        curr = curr.next;
    }
    return sentinel.next;
}

原题链接

移除链表元素