给定一个链表,删除链表的倒数第 n 个节点并返回链表的头指针
例如,
给出的链表为: 1\to 2\to 3\to 4\to 51→2→3→4→5, n= 2n=2.
删除了链表的倒数第 nn 个节点之后,链表变为1\to 2\to 3\to 51→2→3→5.
数据范围: 链表长度 0\le n \le 10000≤n≤1000,链表中任意节点的值满足 0 \le val \le 1000≤val≤100
要求:空间复杂度 O(1)O(1),时间复杂度 O(n)O(n)
备注:
题目保证 nn 一定是有效的
1.不加头结点
ListNode* removeNthFromEnd(ListNode* head, int n) {
// write code here
ListNode* fast = head;
ListNode* slow = head;
if(!fast)//如果链表为空,直接退出
return NULL;
for(int i=0;i<n;i++){//fast在倒数n个点
fast=fast->next;// 题目备注:题目保证 n 一定是有效的,所以不用考虑n大于表长
}
if(fast == nullptr)//如果表长=n,那么刚好删除第一个数,没有这一句那就需要加一个头结点res
return head->next;
ListNode* pre=head;
while(fast){
fast=fast->next;
pre=slow;//pre记录slow上一时刻的位置
slow=slow->next;
}
pre->next=slow->next;
return head;
}
这个里可以改进,不用pre
ListNode* pre=head;
while(fast){
fast=fast->next;
pre=slow;//pre记录slow上一时刻的位置
slow=slow->next;
}
pre->next=slow->next;
改进后
while(fast->next){
fast=fast->next;
slow=slow->next;
}
slow->next=slow->next->next;
2.加头结点
ListNode* removeNthFromEnd(ListNode* head, int n) {
// write code here
ListNode* fast = head;
ListNode* slow = head;
ListNode* res = new ListNode(-1);//创建一个虚拟头结点,当要删除第一个元素时,重要作用
res->next=head;
ListNode* pre=res;//当要删除第一个结点,pre->next=slow->next;
if(!fast)//如果链表为空,直接退出
return NULL;
for(int i=0;i<n;i++){//fast在倒数n个点
fast=fast->next;// 题目备注:题目保证 n 一定是有效的,所以不用考虑n大于表长
}
while(fast){
fast=fast->next;
pre=slow;
slow=slow->next;
}
pre->next=slow->next;
return res->next;
}