Hot100-Day39-T19删除链表的倒数第N个节点

0 阅读2分钟

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 <= 30
  • 0 <= Node.val <= 100
  • 1 <= 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;
    }
};