题目描述
给你一个链表,删除链表的倒数第 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]
在本题目中有可能出现删除头结点的情况,因此需要设置虚拟头结点,设置快慢指针,快指针先走 n 步,然后慢指针再向后移动,由于慢指针是从虚拟头结点,因此快指针结束的时候慢指针指向的是倒数第 n-1 个节点的位置,如果慢指针的下面的next的next位置上有值,那么直接拼上,反之说明删除的是最后一个数字,因此直接拼接上nullptr即可。
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode* dummpy=new ListNode(-1);
dummpy->next=head;
ListNode* fast=head;
ListNode* slow=dummpy;
while(n-- && fast) fast=fast->next;
while(fast)
{
slow=slow->next;
fast=fast->next;
}
if(slow->next->next) slow->next=slow->next->next;
else slow->next=nullptr;
return dummpy->next;
;
}
};