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

题目描述

给你一个链表，删除链表的倒数第 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

代码

/**
 * 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) {
        // 创建一个虚拟节点 dummy，指向 head
        ListNode *dummy = new ListNode(0);
        dummy->next = head;

        // 初始化两个指针 p 和 pre，均指向 dummy
        ListNode *p = dummy;
        ListNode *pre = dummy;

        // 计算链表的长度
        int m = 0;
        while(p->next) {
            m++;
            p = p->next;
        }

        // 移动 pre 指针到链表的第 m-n 个节点
        for(int i = 0; i < m - n; i++) {
            pre = pre->next;
        }

        // p 指针指向要删除的节点
        p = pre->next;

        // 删除节点 p
        pre->next = p->next;
        delete p;

        // 返回新的链表头节点
        return dummy->next;
    }
};