算法系列之链表删除节点JS/TS版

链表删除节点

JS/TS的链表节点表示方法

// Definition for singly-linked list.
function ListNode(val, next) {
    this.val = (val === undefined ? 0 : val)
    this.next = (next === undefined ? null : next)
}

// Definition for singly-linked list.
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)
    }
}

从链表中删除一个节点

https://leetcode.cn/problems/remove-linked-list-elements/description/

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

示例：
输入：head = [1,2,6,3,4,5,6], val = 6
输出：[1,2,3,4,5]

输入：head = [], val = 1
输出：[]

思路：
从链表删除一个节点，相信方法大家都懂，就是把它前一个节点的next指针指向它的下一个节点，也就是跳过了这个节点，它无法再被到达，相当于它被删除了。

普通写法

思路简单但是写起来很多条件要判断，特别是头节点需要额外处理，然后要判断各种空节点和单节点的问题。

function removeElements(head: ListNode | null, val: number): ListNode | null {
    if (!head) return null;
    while (head.val == val) {
        head = head.next;
        if (!head) return null;
    }
    if (!head.next) return head;
    let preNode = head; 
    let node = head.next;
    while(node) {
        if (node.val === val) {
            preNode.next = node.next;
        } else {
            preNode = node;
        }
        node = node.next;
    }
    return head;
};

虚拟头节点写法

移除头结点和移除其他节点的操作方式是不一样，上面代码可见，需要单独写一段逻辑来处理移除头结点的情况。
那设置一个虚拟头结点，这样原链表的所有节点就都可以按照统一的方式进行移除了。

function removeElements(head: ListNode | null, val: number): ListNode | null {
    const dummy = new ListNode(0, head);
    let preNode = dummy; 
    let node = head;
    while(node) {
        if (node.val === val) {
            preNode.next = node.next;
        } else {
            preNode = node;
        }
        node = node.next;
    }
    return dummy.next;
};