题目链接

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;}
};