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

四、基本思路

使用两个指针，p：一个指向第一个元素，q：另一个指向之后的第n个元素。保持两者的距离，遍历链表。当q的下一个元素为空（即遍历结束）时，p的下一个元素即为要删除的元素。

注意：

1）要考虑将删除的元素为第一个元素

2）题目规定n小于等于结点数目，所以要删除的元素必存在（此题无需考虑这种特殊情况）

五、代码

/**
 * 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 *L = head, *p = head, *q = p;
        for (int i = 0; i < n; i++) {//找到p之后的第n个结点
            q = q->next;
        }
        if (q == nullptr) {//要删除的元素为链表的第一个元素
            return L->next;
        }
        while (q->next != nullptr) {
            p = p->next;
            q = q->next;
        }
        p->next = p->next->next;//删除倒数第 n 个结点
        return L;
    }
};