题目描述:
给你一个链表,删除链表的倒数第 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]
解析:
这题实际上算比较简单的,主要是思考如何标记倒数第n个节点的前序节点。自然,可以先遍历一次获取总长度然后再靠一轮循环获取,不过这显然不合题目的要求。这里主要还是记录一下各种思路为主。
方法一:依靠栈
由于这里要求倒数的第n个节点,若这是个双向链表那想必是十分轻松的。反过来思考,之所以双向链表轻松是因为在遍历链尾时,可靠前序指针实现“回退”操作,再此应该能联想到栈的先进后出的特性。
此外,在这里还需要注意若删除的节点是第一个节点的情况,即n为链表的长度。此时可在第一个节点前安插一个空节点作为头结点实现所有节点操作一致。
代码:
#include <iostream>
#include <stack>
using namespace std;
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* fake = new ListNode(0, head);
//临时指针,由于指向后续节点
ListNode* temp = fake;
//栈
stack<ListNode*> my_stack;
//当前指针不是空指针时
//压入所有的节点
while (temp)
{
my_stack.push(temp);
temp = temp->next;
}
//弹出n个节点,此时栈顶为倒数第n+1个节点,
//即倒数第n个节点的前序节点
for(int i=0;i<n;i++)
{
my_stack.pop();
}
//创建指针指向栈顶
ListNode* prev = my_stack.top();
prev->next = prev->next->next;
//由于开始时建立了一个空节点(头结点),因此应返回其后继节点
ListNode* ans = fake->next;
//new出来的堆记得释放
delete fake;
return ans;
}
};
(碎碎念:嘛,严格来说这是不是靠一趟扫描还是有点虚的,毕竟出栈还是多了n个步骤)
方法二:滑动窗口
我们可设计一个滑动窗口,使得:当窗口末端为链尾的空节点时,窗口的前端指向倒数第n个节点,而前后端的距离差距,可直接得为n。
代码:
//空节点
ListNode* fake = new ListNode(0, head);
//滑动窗口的前后端
ListNode* first = fake;
ListNode* second = head;
//后端先走n格拉长窗口至n
for(int i=0;i<n;i++)
{
second = second->next;
}
//当窗口后端不为空时,整个窗口向后移动
//结束循环后窗口前端为倒数第n+1个节点
while (second)
{
second = second->next;
first = first->next;
}
first->next = first->next->next;
ListNode* ans = fake->next;
delete fake;
return ans;