轻松掌握！链表中倒数第 N 个节点的删除技巧

前端

2023-03-11 11:19:14

删除链表中的倒数第 N 个节点：双指针法详解

什么是双指针法？

双指针法是一种链表算法技术，它利用两个或更多指针同时遍历链表，以高效地解决各种链表问题。在删除倒数第 N 个节点的场景中，我们使用两个指针：

慢指针（slow）： 从链表头结点开始，每次移动一步。
快指针（fast）： 也从链表头结点开始，但先移动 N 步，然后再与慢指针一起移动。

如何删除倒数第 N 个节点？

删除链表中的倒数第 N 个节点涉及以下步骤：

移动快指针 N 步： 将快指针向前移动 N 步，到达链表的第 N+1 个节点。
同时移动慢指针和快指针： 当快指针到达链表尾部时，慢指针指向的就是倒数第 N 个节点的前一个节点。
删除目标节点： 将慢指针指向的目标节点的前一个节点的 next 指针指向目标节点的 next 指针，即可删除目标节点。

代码示例

Python：

def remove_nth_from_end(head, n):
    slow = head
    fast = head
    
    # 移动快指针 N 步
    for _ in range(n):
        fast = fast.next

    # 如果快指针已经到达链表尾部，则删除头结点
    if not fast:
        return head.next

    # 同时移动慢指针和快指针，直到快指针到达链表尾部
    while fast.next:
        slow = slow.next
        fast = fast.next

    # 删除目标节点
    slow.next = slow.next.next

    return head

Java：

public ListNode removeNthFromEnd(ListNode head, int n) {
    ListNode slow = head;
    ListNode fast = head;

    // 移动快指针 N 步
    for (int i = 0; i < n; i++) {
        fast = fast.next;
    }

    // 如果快指针已经到达链表尾部，则删除头结点
    if (fast == null) {
        return head.next;
    }

    // 同时移动慢指针和快指针，直到快指针到达链表尾部
    while (fast.next != null) {
        slow = slow.next;
        fast = fast.next;
    }

    // 删除目标节点
    slow.next = slow.next.next;

    return head;
}

C++：

ListNode* removeNthFromEnd(ListNode* head, int n) {
    ListNode* slow = head;
    ListNode* fast = head;

    // 移动快指针 N 步
    for (int i = 0; i < n; i++) {
        fast = fast->next;
    }

    // 如果快指针已经到达链表尾部，则删除头结点
    if (fast == nullptr) {
        return head->next;
    }

    // 同时移动慢指针和快指针，直到快指针到达链表尾部
    while (fast->next != nullptr) {
        slow = slow->next;
        fast = fast->next;
    }

    // 删除目标节点
    slow->next = slow->next->next;

    return head;
}