Leetcode-19: Remove Nth Node From End of List

最新推荐文章于 2024-09-05 06:26:26 发布

原创最新推荐文章于 2024-09-05 06:26:26 发布 · 139 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Leetcode

algorithm-design 专栏收录该内容

817 篇文章

订阅专栏

本文介绍了一种使用双指针技术来高效地从链表中删除倒数第N个节点的方法。首先两个指针同时指向链表头结点，第一个指针向前移动N+1次后开始带动第二个指针一起移动，直至第一个指针到达链表尾部，此时第二个指针正好位于待删除节点的前一个位置。

我的方法就是用2个指针，一开头都指向head。当第1个指针往后移动到n+1的时候，带动第2个指针往后移动。当第1个指针到尾后，第2个指针就指向要删掉的那个节点的前一个节点。
注意:
1) 有可能刚好要删掉的那个节点是head (acc==n的情况)。

#include <iostream>

using namespace std;

struct ListNode {
    int val;
    ListNode *next;
    ListNode(int x) : val(x), next(NULL) {}
};


ListNode* removeNthFromEnd(ListNode* head, int n) {
    if (!head) return NULL;

    int acc=0;
    ListNode* p1=head;
    ListNode* p2=head;

    while(p1) {
        if (acc++>n) p2=p2->next;
        p1=p1->next;
    }
    //here p2 actually points to the previous node of the deleted node.
    if (acc==n){
        ListNode* temp=head;
        head=head->next; temp->next=NULL; return head;
    }else{
        ListNode* temp=p2->next;
        p2->next=p2->next->next;
        temp->next=NULL;
        return head;
    }

}

int main()
{
    ListNode a(1),b(2),c(3),d(4),e(5);
    a.next=&b;
    b.next=&c;
    c.next=&d;
    d.next=&e;

    ListNode*p=removeNthFromEnd(&a, 2);
    while(p){
        cout<<p->val<<" ";
        p=p->next;
    }

    return 0;
}