203. 移除链表元素

题目

🔗题目链接：203. 移除链表元素 - 力扣（LeetCode）

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

思路

• 使用虚拟头节点。新创建一个虚拟头节点，让虚拟头节点的 next 指向 head，这样遍历的时候就可以直接开始，不需要单独考虑第一个节点。
• 正常遍历。这种需要对头节点单独进行处理，不推荐

代码

class ListNode {
  val: number;
  next: ListNode | null;
  constructor(val?: number, next?: ListNode | null) {
    this.val = val === undefined ? 0 : val;
    this.next = next === undefined ? null : next;
  }
}

function removeElements(head: ListNode | null, val: number): ListNode | null {
  if (!head) return null;

  let dummyHead = new ListNode(undefined, head);
  let pointer = dummyHead;

  while (pointer.next) {
    if (pointer.next.val === val) {
      let temp: ListNode | null = pointer.next;

      pointer.next = pointer.next.next;
      
      // 被删除的链表节点不会指向下一个节点
      temp.next = null;
      // 创建的临时变量设为 null，GC
      temp = null;
    } else {
      pointer = pointer.next;
    }
  }

  return dummyHead.next;
}