常规方法,遍历两次不讲,只讲如何扫描一次完成删除
利用双指针+虚拟头结点
第一个指针first先单独前移n-1位(注意不是n位,自己理解一下),移动完n-1位之后,cur指针开始与first指针同时移动
(这里为了完成接下去的删除操作,所以使用了pre保存cur的前一个节点)
指针一直移动,之后first到达最后一个节点,此时cur指针指向的位置就是倒数第n个节点,删除之
使用虚拟头结点,可以避免讨论cur为头结点时,pre不存在的情况
尤其注意最后的返回值为dummyHead->next,而不是head,这是为什么呢?
因为所要删除的元素有可能就是head,那么如果返回head,就会出现指针指向的异常,而dummyHead不存在与真实的链表删除序列中,不会成为被删除的节点,避免了这种情况。这就是利用虚拟头结点的好处,请认真体会
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode* dummyHead = new ListNode(0)//设置虚拟头结点
dummyHead->next = head;
ListNode* first = head;
ListNode* cur = head;
ListNode* pre = dummyHead;
for(int i = 1; i < n; i++)
first = first->next;
while(first->next){
cur = cur->next;
pre = pre->next;
first = first->next;
}
pre->next = cur->next;
cur->next = NULL;
delete cur;
return dummyHead->next;
}
};