19. 删除链表的倒数第 N 个结点 - 力扣(LeetCode)
一、题目要求
给你一个链表,删除链表的倒数第 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
二、解法1-递归 O(N)
递归到nullptr后就返回上一个递归时的cur是倒数第几个节点,找到倒数第N个节点的上一个节点,然后去掉第N个节点,如果是头节点需要单独处理。
class Solution {
public:
int _removeNthFromEnd(ListNode* cur) {
if (cur == nullptr)
return 1;
int n = _removeNthFromEnd(cur->next);
if (cur == *Head && m == n) { // 删除头节点
*Head = (*Head)->next;
}
if (m == n - 1) {
cur->next = cur->next->next;
}
return n + 1;
}
ListNode* removeNthFromEnd(ListNode* head, int n) {
m = n;
Head = &head;
_removeNthFromEnd(head);
return head;
}
private:
int m;
ListNode** Head;
};
三、解法2-双指针 O(N)
我们可以让两个指针first和last先相距n,然后first和last同时向后走,当last走到nullptr时停止;
此时first的位置就是要删除的节点,所以还要一个prev保存first的前一个节点,便于删除;
同样的,删除头节点要作特殊处理。
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
if(head->next == nullptr)
return nullptr;
ListNode* first = head;
ListNode* last = head;
while(n--)
{
last = last->next;
}
ListNode* prev = first;
while(last)
{
last=last->next;
prev = first;
first=first->next;
}
if(head == first)
head = head->next;
prev->next = prev->next->next;
return head;
}
};