题目链接

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

题目描述

给你一个链表，删除链表的倒数第 $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\le sz\le 30$
• $0\le Node.val\le 100$
• $1\le n\le sz$

解法一：栈

用一个栈 $stk$，先将所有链表节点入栈。

然后再弹出 $n$ 个节点，此时栈顶节点就是 倒数第 $n+1$ 个节点。

我们直接让 倒数第 $n+1$ 个节点 指向 倒数第 $n-1$ 个节点，这样就把 倒数第 $n$ 个节点删除了。

注意： 由于可能弹出全部的节点，栈可能为空，所以我们要对边界条件做特殊处理。

时间复杂度： $O(n)$

C++代码：

class Solution {
public:
    ListNode* removeNthFromEnd(ListNode* head, int n) {
        stack<ListNode*> stk;

        ListNode* cur = head;
        while(cur){
            stk.push(cur);
            cur = cur->next;
        }

        for(int i = 0;i < n;i++) stk.pop();
        
        if(stk.empty()) return head->next;
        ListNode* pre = stk.top();
        pre->next = pre->next->next;

        return head;
    }
};

解法二：双指针

我们开始让 $fast,slow$ 分别指向 链表头节点 $head$；

接着让 $fast$ 先走 $n$ 步，此时 $fast$ 和 $slow$ 之间就间隔 $n$ 个节点。

接着再让 $fast,slow$ 往后移动，当 $fast$ 为空的时候说明此时 $slow$ 就是 倒数第 $n$ 个节点。

我们在删除这个 倒数第 $n$ 个节点即可。

时间复杂度： $O(n)$

C++代码：

class Solution {
public:
    ListNode* removeNthFromEnd(ListNode* head, int n) {
        
        ListNode* dummy = new ListNode(-1);
        ListNode* pre = dummy;
        pre->next = head;

        ListNode* fast = head , *slow = head;
        while(n--){
            fast = fast->next;
        }

        while(fast){
            pre = pre->next;
            slow = slow->next;
            fast = fast->next;
        }

        pre->next = pre->next->next;

        return dummy->next;

        return head;
    }
};