Day39[26/3/8]T19删除链表的倒数第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 <= 300 <= Node.val <= 1001 <= n <= sz
进阶:你能尝试使用一趟扫描实现吗?
解题思路
用快慢指针,其中快指针先出发,先走 n 步,然后快慢指针一起走,那么快指针永远比慢指针快 n 步,所以当快指针为 nullptr 时,慢指针刚好指向倒数第 n 个节点。
但是呢,你要考虑到这道题的坑,比如,想要删除第 n 个节点,你应当是处于倒数第 n+1 个节点才好操作,此外,还有万一你要删除的就是开头的第一个节点呢?
所以可以这样,你在头节点外,再加一个傀儡节点,慢指针先指向这个傀儡节点,快指针指向正常的头节点,这样慢指针比快指针慢 n+1 步,并且就算是你需要删除原本的头节点,但是你现在处理起来就和删除中间节点是一样的。
此外还有一个小细节,创建傀儡节点时,建议不要用 new(也就是在堆上创建),而是在栈上创建,因为这样的话,你退出程序时,这个变量会被自动析构,不需要手动 delete 也就不会内存泄漏了!
Code
#include <iostream>
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(0, head); // 假头,防止需要修改head的情况
ListNode *fast = head; // 快指针
ListNode *slow = &dummyHead; // 慢指针
// 1. 快指针先走n步
for (int i = 0; i < n; i++)
{
fast = fast->next;
}
// 2. 然后一起走
while (fast != nullptr)
{
fast = fast->next;
slow = slow->next;
}
// 3. 删除节点
ListNode *temp = slow->next;
slow->next = slow->next->next;
delete temp;
temp = nullptr;
return dummyHead.next;
}
};