LeetCode 19.删除链表的倒数第 N 个结点

原题链接

难度：$middle$

题目描述

给你一个链表，删除链表的倒数第 $n$ 个结点，并且返回链表的头结点。

示例 1：

输入：head = [1,2,3,4,5], n = 2
输出：[1,2,3,5]

示例 2：

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

示例 3：

输入：head = [1,2], n = 1
输出：[1]

提示：

• 链表中结点的数目为 $sz$
• $1<=sz<=30$
• $0<=Node.val<=100$
• $1<=n<=sz$

进阶： 你能尝试使用一趟扫描实现吗？

---

算法

1. 题目中给的是倒数第几个结点，如果找出倒数的第几个数字？
2. 删除结点是在原链表上删除还是更新在新的链表上？

自己的想法：链表是否需要翻转然后删除第几个结点 （×） 太麻烦

结论：可分为两次遍历和一次遍历。

(两次遍历)

链表题目，头节点特判。使用虚拟头节点。

1. 第一次遍历求出链表长度。
2. 第二次遍历删掉指定结点。
3. 在 head 前新增 dummy 节点的方式避免删除首结点时出现问题。

C++ 代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* removeNthFromEnd(ListNode* head, int n) {
        auto dummy = new ListNode(-1);
        dummy->next = head;

        int list_size = 0;
        for (auto p = dummy; p; p = p->next) list_size ++;

        auto p = dummy;
        for (int i = 0; i < list_size - n - 1; i ++) p = p->next;
        p->next = p->next->next;

        return dummy->next;
    }
};

(一次遍历)

1. 一次遍历的思路是设置两个指针，让其中一个指针先走 n 步，然后两个指针再同时走。当第二个指针走到最后时，第一个指针的位置就是需要被删除的节点。
2. 在 head 前添加 dummy 节点，避免出现边界问题。
3. 设置两个指针 $p_1$ 和 $p_2$，均初始化为 dummy 节点。
4. $p_2$ 指针先移动 n 次，然后 $p_1$ 和 $p_2$ 指针同时向后移动，直到 $p_2$ 指向最后一个节点。
5. 此时 $p_1$ 结点指向的下一个结点需要删除。

C++ 代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* removeNthFromEnd(ListNode* head, int n) {
        ListNode *dummy = new ListNode(0, head);

        auto *p1 = dummy, *p2 = dummy;

        while(n --) 
            p2 = p2->next;

        while(p2->next) {
            p1 = p1->next;
            p2 = p2->next;
        }

        p1->next = p1->next->next;

        return dummy->next;
    }
};