快慢指针—删除链表的倒数第N个节点（leetcode 19）

题目描述

给你一个链表，删除链表的倒数第 n 个结点，并且返回链表的头结点。

进阶：你能尝试使用一趟扫描实现吗？

 

示例 1：

输入：head = [1,2,3,4,5], n = 2
输出：[1,2,3,5]

示例 2：

输入：head = [1], n = 1
输出：[]

示例 3：

输入：head = [1,2], n = 1
输出：[1]

 

提示：

    链表中结点的数目为 sz
    1 <= sz <= 30
    0 <= Node.val <= 100
    1 <= n <= sz

算法分析

双指针

我们也可以在不预处理出链表的长度，以及使用常数空间的前提下解决本题。

由于我们需要找到倒数第 n个节点，因此我们可以使用两个指针 first 和 second同时对链表进行遍历，并且 first 比 second超前 n个节点。当 first遍历到链表的末尾时，second就恰好处于倒数第 n个节点。

具体地，初始时 first 和 second均指向头节点。我们首先使用 first对链表进行遍历，遍历的次数为 n。此时，first和 second之间间隔了 n−1个节点，即 first比 second超前了 n个节点。

在这之后，我们同时使用 first 和 second对链表进行遍历。当 first遍历到链表的末尾（即 first为空指针）时，second恰好指向倒数第 n个节点。

根据方法一和方法二，如果我们能够得到的是倒数第 n个节点的前驱节点而不是倒数第 n 个节点的话，删除操作会更加方便。因此我们可以考虑在初始时将 second指向哑节点，其余的操作步骤不变。这样一来，当 first遍历到链表的末尾时，second的下一个节点就是我们需要删除的节点。

代码

/**
 * 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* dummy = new ListNode(0, head);
        ListNode* first = head;
        ListNode* second = dummy;
        for(int i = 0; i < n; ++i) {
            first = first->next;
        }
        while(first) {
            second = second->next;
            first = first->next;
        }

        ListNode* tmp = second->next;
        second->next = second->next->next;
        delete tmp;
        ListNode* ans = dummy->next;
        delete dummy;
        return ans;
    }
};

时间复杂度分析

• 时间复杂度：O(L)，其中 L是链表的长度。

• 空间复杂度：O(1)。