给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
思路:
此题的简化版本就是找出链表的倒数结点,快慢指针即可。此题还需删除该结点就有些细节需要注意一下。
1.与直接返回倒数结点不同,删除结点还需知道指向待删除结点的指针。快指针先走n+1个结点。
两指针同时移动直至快指针为空,此时慢指针指向待删结点前一个结点。
2.此时考虑程序的鲁棒性。当n=5时,待删结点为第一个结点,当快指针还为移动n个时已经为空
且无法直接同上操作一样删除该结点。因此应判断待删除结点是否为头指针指向结点。很自然的想到可以增加头结点来同一操作,因为所给链表不含真正意义上的头结点,因此得到输入后人为增加一个伪头结点。
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode* dummy=new ListNode(0);
dummy->next=head;//与原链表相连
ListNode *p=dummy;
ListNode *q=dummy;
for(int i=0;i<=n;i++)
p=p->next;
while(p)
{
p=p->next;
q=q->next;
}
p=q->next;
q->next=p->next;
delete p;
return dummy->next;//注意这里返回的是伪头结点下一个结点
}
};