LeetCode|链表|19. 删除链表的倒数第 N 个结点

一、题目描述

给你一个链表，删除链表的倒数第 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]

二、解题思路

        删除节点，无非就是要找到这个节点的前一个节点，然后删除。所以核心是，如何找到倒数第n个节点的前一个节点？

        定义快慢指针，slow和fast，令其都指向虚拟头结点。让fast先往后移动n步，然后slow和fast同时往后移动，直到fast指针指向最后的NULL，此时slow就指向了倒数第n个节点。

        那么想要找到倒数第n个节点的前一个节点，和上面同理。只不过这里是让fast先移动n+1步。完成上述步骤后，slow指向的就是要删除节点的前一个节点了。

        ps：要注意的是，fast先往后移动n+1步的过程中，判断条件里要加fast != NULL

三、代码实现

/**
 * 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* dummyHead = new ListNode(0);//new一个虚拟头结点
        dummyHead->next = head;
        //定义快慢指针，找到倒数第n个节点的前一个节点
        ListNode* fast = dummyHead;
        ListNode* slow = dummyHead;
        //让快指针fast先走n+1步
        int step = n + 1;
        while(step-- && fast != NULL){
            fast = fast->next;
        }
        //两个指针同时移动，直到fast指向null，此时slow指向要删除的节点的前一个节点
        while(fast != NULL){
            fast = fast->next;
            slow = slow->next;
        }
        ListNode* temp = slow->next;//临时指针指向要删除的节点
        slow->next = slow->next->next;//删除节点
        head = dummyHead->next;
        //释放内存
        delete dummyHead;
        delete temp;
        return head;
    }
};