19. 删除链表的倒数第 N 个结点 - 力扣(LeetCode)
题目描述
给你一个链表,删除链表的倒数第 n
个结点,并且返回链表的头结点。
样例输入
输入:head = [1,2,3,4,5], n = 2 输出:[1,2,3,5]
示例 2:
输入:head = [1], n = 1 输出:[]
示例 3:
输入:head = [1,2], n = 1 输出:[1]
题解
如何删除链表中的节点
假如要删除的结点指针为p,在链表中我们需要找到指针p的前一个结点q,之后通过q->next=p->next删除节点
因此问题转换为找到倒数第N个节点的前一个节点
如何找到删除节点前一个的位置
设想有双指针 p 和 q ,那么当 q 指向尾后元素 NULL,而此时p 与 q 之间相隔的元素个数为 n时 ,我们就找到了删除元素的前一个节点的位置,此时删除掉 p 的下一个指针就完成了要求。
故算法步骤可设置为:
- 设置指针p和q,初始时都指向头结点
- 先让q走n+1步,之后q和p同时向后移动,此时便保持了q和p之间的距离
- 此后让q一直向后移动,当q指向尾后节点时,p就找到了要删除节点的前一个节点的位置
细节处理(头结点的删除)
如果题目中要求我们删除第一个节点(此时n等于链表的长度),那么我们就要删除头结点,而为了方便的删除头结点,我们需要创建一个虚拟节点指向链表的头结点,返回时返回该虚拟节点的下一个节点即可
思路整理
- 设置虚拟节点 dummyHead 指向 head(为了头结点的删除)
- 设定双指针 p 和 q,初始都指向虚拟节点 dummyHead
- 移动 q,直到 p 与 q 之间相隔的元素个数为 n(保持指针q与指针q的间隔)
- 同时移动 p 与 q,直到 q 指向的为 NULL
- 将 p 的下一个节点指向下下个节点
代码
/**
* 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* dummyHead=new ListNode(0);
dummyHead->next=head;//虚拟头结点,用于删除头结点
ListNode* q=dummyHead,*p=dummyHead;
n+=1;
//让q先走n+1步
while(n--)
q=q->next;
//之后q和p同时移动,直到q走到尾后指针的位置,此时p指向的位置就是我们要找的位置
while(q)
{
q=q->next;
p=p->next;
}
//执行删除操作
ListNode* e=q->next;
q->next=e->next;
delete e;
return dummyHead->next;
}
};