https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list/
给你一个链表,删除链表的倒数第 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 <= sz <= 30
0 <= Node.val <= 100
1 <= n <= sz
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
找到倒数第 N 个节点
- 要删除倒数第 N 个节点,第一步要先找到倒数第 N 个节点(实际上对于单向链表,还需要找到倒数第 N+1 个节点才能完成删除倒数第 N 个节点的操作)。
- 简单的思路是,遍历确定链表长度 L,通过计算得到倒数第 N 个节点是顺数第 L-N 个节点。
- 快慢指针的思路是,定义两个指针,快指针先走 N 步,剩下的路程自然是 L-N,这时再让快慢指针同时走,等快指针走路末尾时,慢指针走过的路程刚好是 L-N,也就是刚好走到倒数第 N 个节点。
使用 dummy 节点来避免单独处理头节点
假如要删除的节点刚好是头节点的话,要额外处理就很麻烦。但如果事先在头节点前面添加一个 dummy 节点,就可以把头节点当成一个普通节点来处理了,这是链表题中常用的技巧。
- 时间复杂度:$O(N)$。
- 空间复杂度:$O(1)$。
C++ Code
/**
* 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:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode *fast = head, *slow = head;
while (n--) {
fast = fast->next;
}
// 额外处理头部节点
if (!fast) {
ListNode *new_head = head->next;
head->next = nullptr;
return new_head;
}
while (fast->next) {
fast = fast->next;
slow = slow->next;
}
ListNode *target = slow->next;
slow->next = target->next;
target->next = nullptr;
return head;
}
};
使用 dummy 节点
/**
* 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:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode *dummy = new ListNode(0, head);
ListNode *fast = dummy, *slow = dummy;
// 快指针先走 n 步
while (n-- > 0) {
fast = fast->next;
}
// 当快指针走到尾节点时
// 慢指针刚好走到倒数 n+1 个节点
while (fast && fast->next) {
fast = fast->next;
slow = slow->next;
}
ListNode *target = slow->next;
slow->next = target->next;
target->next = nullptr;
return dummy->next;
}
};