链接:删除链表的倒数第N个结点
描述:给你一个链表,删除链表的倒数第 n 个结点,并且返回链表的头结点。
进阶:你能尝试使用一趟扫描实现吗?
示例:
输入:head = [1,2,3,4,5], n = 2
输出:[1,2,3,5]
思路:看到题目最开始想到的就是先扫描一遍确定链表长度len,再顺序遍历链表删除第len-n+1的结点。显然这样做并不能满足一次遍历。
代码1(先确定长度):
/**
* Definition for singly-linked list.
* 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:
int getLength(ListNode* head)
{
int length = 0;
while(head)
{
length++;
head = head->next;
}
return length;
}
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode* dummy = new ListNode(0,head);
int length = getLength(head);
ListNode* cur = dummy;
for(int i = 0; i < length - n + 1 ; i++)
{
cur = cur->next;
}
cur->next = cur->next->next;
ListNode* ans = dummy->next;
delete dummy;
return ans;
}
};
与此同时我想到栈“后进先出”的特点,可以在一次遍历之时pop出倒数第n个结点。
代码2(栈):
class Solution{
public:
ListNode* removeNthFromEnd(ListNode* head, int n)
{
ListNode* dummy = new ListNode(0, head);
stack<ListNode*> stk;
ListNode* cur = dummy;
while(cur)
{
stk.push(cur);
cur = cur->next;
}
for(int i = 0; i < n ; i++)
{
stk.pop();
}
ListNode* prev = stk.top();
prev->next = prev->next->next;
ListNode* ans = dummy->next;
delete dummy;
return ans;
}
};
这个时候看了题解和评论区,发现快慢双指针是一个很不错的解法。算法原理是:让一快一满两个指针在同一结点出发,快指针永远先于满指针n步。一开始快指针出发,行动n个结点后,慢指针跟着出发,当快指针运动到链表尾巴的时候,慢指针就找到了倒数第n个结点了。
代码3(快慢双指针):
class Solution{
public:
ListNode* removeNthFromEnd(ListNode* head, int n)
{
ListNode* dummy = new ListNode(0, head);
ListNode* first = head;
ListNode* second = dummy;
for(int i = 0; i < n ; i++)
{
first = first->next;
}
while(first)
{
first = first->next;
second = second->next;
}
second->next = second->next->next;
ListNode* ans = dummy->next;
delete dummy;
return ans;
}
};
btw,蓝桥杯快要来了,大一学的数据结构知识有点忘了,在此复习一下,顺便祝大家取得一个好成绩😀