题目:
给你一个链表,删除链表的倒数第 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]
思路:
- 首先涉及到添加或删除节点,需要考虑虚拟头节点,保证真实头节点和后续节点操作一致
- 题目要求是倒数第n个节点,难点在于单链表指向是正向的,要如何正向找到倒数第n个节点。
步骤:
- 定义fast指针和slow指针,初始值为虚拟头结点,如图:
- fast首先向后移动n位 ,因为只有这样同时移动的时候slow才能指向删除节点的上一个节点(方便做删除操作)。如图:
- fast和slow同时移动,直到fast为尾节点,如图:
详细解释如下:
节点总数:链表节点数+虚拟头节点 = size + 1
从虚拟头节点到尾节点移动步数:size
fast先移动n步,然后fast和slow同时移动size - n步后,fast指向尾节点,slow指向倒数第n + 1节点。 - 删除slow指向的下一个节点。
代码:
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode* dummyHead = new ListNode(0); // 定义虚拟头节点,保证真实头节点和后续节点操作一致
dummyHead->next = head;
ListNode* slow = dummyHead;
ListNode* fast = dummyHead;
//将fast向后移动n位,这样后续同时移动fast和slow,当fast为最后一个节点时,slow指向倒数第n个节点的上一节点
while(n-- && fast != nullptr) {
fast = fast->next;
}
// 当fast为最后一个节点时,slow指向倒数第n个节点的上一节点
while(fast->next != nullptr) {
fast = fast->next;
slow = slow->next;
}
// 删除倒数第n个节点
ListNode* tmp = slow->next;
slow->next = slow->next->next;
delete tmp;
// 删除虚拟头节点
ListNode* tmp1 = dummyHead;
head = dummyHead->next;
delete tmp1;
return head;
}
};
总结:
时间复杂度: O(n)
空间复杂度: O(1)
- 首先涉及到添加或删除节点,需要考虑虚拟头节点,保证真实头节点和后续节点操作一致
- 题目要求是倒数第n个节点,难点在于单链表指向是正向的,要如何正向找到倒数第n个节点。
参考:
代码随想录