给你一个链表,删除链表的倒数第 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]
// 无注释版
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 *dummyHead=new ListNode(0);
ListNode *p=dummyHead;
ListNode *q=dummyHead;
dummyHead->next=head;
for(int i=0;i<=n;i++)
q=q->next;
while(q!=NULL) {
p=p->next;
q=q->next;
}
ListNode *tmp=p->next;
p->next=p->next->next;
tmp=dummyHead->next;
delete dummyHead;
return tmp;
}
};
// 注释版
// 链表的定义
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) { // 通过参数传入给定数据
// 创建头结点dummyHead
// 创建指向头结点的指针p和q
ListNode *dummyHead=new ListNode(0);
ListNode *p=dummyHead;
ListNode *q=dummyHead;
// 让头结点的指针指向链表第一个结点
dummyHead->next=head;
// 让指针q相距指针p n 个结点的距离
for(int i=0;i<=n;i++)
q=q->next;
//让指针p和指针q同时遍历链表,q指向空结束
// 此时p指向倒数第n+1个结点
while(q!=NULL) {
p=p->next;
q=q->next;
}
// 删除倒数第n个结点
ListNode *tmp=p->next;
p->next=p->next->next;
// 让tmp指向第一个结点,然后释放头结点
tmp=dummyHead->next;
delete dummyHead;
// 返回第一个结点
// 不是返回head,head不一定是第一结点
return tmp;
}
};