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

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]

提示：

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

解题思路

双指针的一个使用

• 设置虚拟头结点，主要目的是方便删除操作

  ListNode* dummyHead = new ListNode();

• 使用快慢指针方法，定义两个指针，让两个指针都指向虚拟头结点

  ListNode* fast = dummyHead

  ListNode* slow = dummyHead

• 让fast指针移动n+1步，只有这样同时移动的时候slow才能指向删除结点的上一个结点，比较想不出来

  while (n-- && fast != NULL) {

  ​ fast = fast->next;

  }

  fast = fast->next;

• fast slow 同时移动，直到fast指向末尾

• 删除slow指向的下一个节点

/**
 * 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);
        dummyHead->next = head;
        ListNode* fast = dummyHead;
        ListNode* slow = dummyHead;
        while (n-- && fast != NULL) {
            fast = fast->next;
        }
        fast = fast->next; // 多走一步
        while (fast != NULL) {
            fast = fast->next;
            slow = slow->next;
        }
        
        slow->next = slow->next->next;

        return dummyHead->next;
    }
};