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

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

思路

先扫描一遍整个链表，记录链表长度cnt，再扫描第二遍，用cnt-n标记要删除的链表元素，注意对删除头结点进行特殊处理，否则访问 pre->next 就会导致空指针解引用错误。

代码

/*** 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* p = head;int cnt = 0;while (p != nullptr) {cnt++;p = p->next;}p = head;int i = 0;ListNode* pre = nullptr;while (p != nullptr) {if (i == cnt - n) {if (pre == nullptr) {// 删除第一个节点，直接将head等于下一个节点head = p->next;} else {pre->next = p->next;}delete (p);break;}pre = p;p = p->next;i++;}return head;}
};