[链表|多解法] leetcode 19 删除链表的倒数第N个节点

[链表|多解法] leetcode 19 删除链表的倒数第N个节点

1.题目

题目链接
给定一个链表，删除链表的倒数第 n 个节点，并且返回链表的头结点。
示例：

给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后，链表变为 1->2->3->5.

说明：
给定的 n 保证是有效的。
进阶：
你能尝试使用一趟扫描实现吗？

2.分析

2.1.直观解法

既然要删除倒数第N个结点，那么统计出链表总共有多少个结点，然后就可以找到倒数第N个结点并删除了。
统计结点个数只需要通过一次遍历：

int getLength(ListNode* head) {
    int length = 0;
    while (head) {
        length++;
        head = head->next;
    }
    return length;
}

然后，我们遍历到第length - (n + 1)个结点时，将该结点的指针指向下下一个结点，便可以达到删除倒数第N个结点的目的。

ListNode* removeNthFromEnd(ListNode* head, int n) {
    ListNode* dummy = new ListNode(0, head);
    int length = getLength(head);
    ListNode* cur = dummy;
    for (int i = 1; i < length - n + 1; ++i) {
        cur = cur->next;
    }
    cur->next = cur->next->next;
    ListNode* ans = dummy->next;
    delete dummy;
    return ans;
}

2.2.栈

栈具有后进先出(LIFO)的特性，因此十分适合用来解决这种涉及到倒数的问题。
我们可以把遍历到的每一个结点都压入栈中，然后弹出的第N个结点便是我们要找的倒数第N结点。
这也可以算是一种用空间换时间的方法。

ListNode* removeNthFromEnd(ListNode* head, int n) {
    ListNode* dummy = new ListNode(0, head);
    stack<ListNode*> stk;
    ListNode* cur = dummy;
    //遍历一次链表,将所有结点压入栈中
    while (cur) {
        stk.push(cur);
        cur = cur->next;
    }
    //找到倒数第N个结点
    for (int i = 0; i < n; ++i) {
        stk.pop();
    }
    //更新倒数第N+1个结点的next指针,完成删除
    ListNode* prev = stk.top();
    prev->next = prev->next->next;
    ListNode* ans = dummy->next;
    delete dummy;
    return ans;
}

2.3.快慢指针法

使用两个指针可以解决许多与链表有关的遍历问题。
对于快指针，我们先让其先走一步，找到链表的顺数第N + 1个结点：

ListNode* fast = head;
while(n){
    fast = fast->next;
    n--;
}

然后，让快慢指针同步前进。由于快指针已经先遍历到了第N + 1个结点，因此当其遍历完毕后，慢指针会还差N步才能遍历完，正好指向倒数第N个结点。

while(fast){
    pre = slow;
    fast = fast->next;
    slow = slow->next; 
}

3.代码

class Solution {
public:
    ListNode* removeNthFromEnd(ListNode* head, int n) {
        ListNode* fast = head;
        ListNode* slow = head;
        ListNode* pre = head;
        while(n){
            fast = fast->next;
            n--;
        }
        while(fast){
            pre = slow;
            fast = fast->next;
            slow = slow->next;  
        }
        if(slow == head){
            return head->next;
        }
        pre->next = slow->next;
        return head;
    }
};