【leetcode】19. 刪除連結串列的倒數第N個節點（remove-nth-node-from-end-of-list）（雙指標）[中等]

連結

https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list/

耗時

解題：8 min
題解：14 min

題意

給定一個連結串列，刪除連結串列的倒數第 n 個節點，並且返回連結串列的頭結點。

思路

要刪除倒數第 n 個節點，需要先找到它前面的那個節點，即倒數第 n+1 個節點，快慢指標可以做到。但需要注意的是，倒數第 n 個節點可能是頭節點，它沒有前一個節點，需要特殊判斷一下。

先將快指標移動 n 次，到達第 n 個節點後面，如果指向空，說明連結串列長度就是 n，要刪除的是頭節點，直接返回頭節點的 next 即可；如果不為空，則將快指標再向後移動一位，此時快慢指標相距 n+1 位，接下來使得快慢指標同時移動，直到快指標指向空，此時慢指標指向倒數第 n+1 個節點，最後將慢指標指向節點的 next 直接指向下下一個節點，完成刪除倒數第 n 個節點。

時間複雜度：$O(length)$

AC程式碼

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* removeNthFromEnd(ListNode* head, int n) {
        ListNode* fast = head;
        ListNode* slow = head;
        for(int i = 0; i < n; ++i) fast = fast->next;
        if(fast == NULL) return head->next;
        fast = fast->next;
        while(fast != NULL) {
            fast = fast->next;
            slow = slow->next;
        }
        slow->next = slow->next->next;
        return head;
    }
};