力扣刷题日记

给定一个链表，删除链表的倒数第 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 个节点。

可以使用快慢指针减少时间复杂度，只需要遍历一次就能找到倒数第 n 个节点。思路如下，定义两个指针，都指向头结点。快指针先走 n 步，然后两个之前一起往前走，当快指针达到链表尾部的时候，满指针指向的就是倒数第 n 个节点。

这里要用到虚拟头结点（就是新定义一个节点，即虚拟头结点，该节点的下一个节点就是题目所给链表的头结点），做链表题经常用到。可以避免链表长度等于 n 时要单独考虑的情况，即删除头结点的情况。

代码如下：

class Solution {
public:
    ListNode* removeNthFromEnd(ListNode* head, int n) {
        ListNode* virtualHead = new ListNode(0, head); //虚拟头结点
        ListNode* fast = head, *slow = virtualHead;
        while(n--) //快指针先走 n 步
            fast = fast->next;
        while(fast){ //快慢指针一起往前走
            fast = fast->next;
            slow = slow->next;
        }
        slow->next = slow->next->next; //删除倒数第 n 个节点
        return virtualHead->next;
    }
};

时间复杂度：O(n)；空间复杂度：O(1)。