删除链表倒数第n个节点

leetcode 题目:
给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。

	给定一个链表: 1->2->3->4->5, 和 n = 2.
	当删除了倒数第二个节点后,链表变为 1->2->3->5.
	note: 
		n>=链表长度时,删除头节点;n<=0时,不做任何操作。

当第一个指针first 比第二个指针领先n步,然后两个指针同步前进,那么第一个指针到达链表尾部时,第二个指针即指向要删除的节点。如图:
删除链表倒数第n个节点_第1张图片

ListNode* removeNthFromEnd(ListNode* head, int n) {
        if(!head || n <= 0) return head;

        ListNode *first = head;
        ListNode *second = head;
        ListNode *prev = NULL;
        for(int i = 0; i < n && first; ++i){
            first = first->next;
        }

        while(first){
            first = first->next;
            prev = second;
            second = second->next;
        }
        if(!prev){ // 删除头结点
        	head = head->next;
        } else {
        	prev->next = second->next;
        }
        delete second;
        return head;
    }

删除prev指针的遍历

令first指针到达链表尾部时,second指针指向删除节点的前一个节点,就可以遍历prev指针了。
使用这种方法,我们需要在链表头添加一个dummy节点,用于处理删除节点为头结点的情况。
删除链表倒数第n个节点_第2张图片

ListNode* removeNthFromEnd(ListNode* head, int n) {
    if(!head || n <= 0) return head;
 2
    ListNode *dummy = new ListNode(0);
    dummy->next = head;

    ListNode *first = dummy;
    ListNode *second = dummy;

    // n >= 链表size,删除头节点
    for(int i = 0; i < n+1 && first; ++i){
        first = first->next;
    }

    while(first){
        first = first->next;
        second = second->next;
    }
    ListNode *prev = second;
    second = second->next;
    prev->next = second->next;
    delete second;
    
    ListNode *ret = dummy->next;
    delete dummy;
    return ret;
}

你可能感兴趣的:(c++,链表,leetcode)