[链表|多解法] leetcode 19 删除链表的倒数第N个节点
1.题目
题目链接
给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
说明:
给定的 n 保证是有效的。
进阶:
你能尝试使用一趟扫描实现吗?
2.分析
2.1.直观解法
既然要删除倒数第N个结点,那么统计出链表总共有多少个结点,然后就可以找到倒数第N个结点并删除了。
统计结点个数只需要通过一次遍历:
int getLength(ListNode* head) {
int length = 0;
while (head) {
length++;
head = head->next;
}
return length;
}
然后,我们遍历到第length - (n + 1)个结点时,将该结点的指针指向下下一个结点,便可以达到删除倒数第N个结点的目的。
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode* dummy = new ListNode(0, head);
int length = getLength(head);
ListNode* cur = dummy;
for (int i = 1; i < length - n + 1; ++i) {
cur = cur->next;
}
cur->next = cur->next->next;
ListNode* ans = dummy->next;
delete dummy;
return ans;
}
2.2.栈
栈具有后进先出(LIFO)的特性,因此十分适合用来解决这种涉及到倒数的问题。
我们可以把遍历到的每一个结点都压入栈中,然后弹出的第N个结点便是我们要找的倒数第N结点。
这也可以算是一种用空间换时间的方法。
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;
}
//找到倒数第N个结点
for (int i = 0; i < n; ++i) {
stk.pop();
}
//更新倒数第N+1个结点的next指针,完成删除
ListNode* prev = stk.top();
prev->next = prev->next->next;
ListNode* ans = dummy->next;
delete dummy;
return ans;
}
2.3.快慢指针法
使用两个指针可以解决许多与链表有关的遍历问题。
对于快指针,我们先让其先走一步,找到链表的顺数第N + 1个结点:
ListNode* fast = head;
while(n){
fast = fast->next;
n--;
}
然后,让快慢指针同步前进。由于快指针已经先遍历到了第N + 1个结点,因此当其遍历完毕后,慢指针会还差N步才能遍历完,正好指向倒数第N个结点。
while(fast){
pre = slow;
fast = fast->next;
slow = slow->next;
}
3.代码
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode* fast = head;
ListNode* slow = head;
ListNode* pre = head;
while(n){
fast = fast->next;
n--;
}
while(fast){
pre = slow;
fast = fast->next;
slow = slow->next;
}
if(slow == head){
return head->next;
}
pre->next = slow->next;
return head;
}
};