【leetcode】19. 删除链表的倒数第N个节点

19. 删除链表的倒数第N个节点

常规方法,遍历两次不讲,只讲如何扫描一次完成删除
利用双指针+虚拟头结点
第一个指针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;
    }
};

猜你喜欢

转载自blog.csdn.net/qq_28038873/article/details/80683833