题目描述
给你一个链表,删除链表的倒数第 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
思路:使用双指针,因为要删除倒数第n个节点,就需要将一个指针,指向这个节点的前一个节点,可以在链表前定义虚拟头节点,这样使得删除节点操作统一,不需要区分是不是头节点;
先让两个指针都指向虚拟头节点,让fast指针先往前走n+1(当fast往前走n+1后,然后fast与slow指针一起往前走,当fast指针指向末尾NULL时,slow指针刚好会指向倒数第n个节点的前一个节点,然后使用slow->next = slow->next->next删除,记住delete掉删除的节点)
完整代码:
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
// 创建一个哑结点,这样可以方便地处理边界情况,如删除头节点
ListNode* dummyHead = new ListNode(0);
dummyHead->next = head;
// 使用两个指针,fast 和 slow,都初始化指向哑结点
ListNode* fast = dummyHead;
ListNode* slow = dummyHead;
// fast 指针先向前移动 n 步,以创建 fast 和 slow 之间的 n 步间隔
while(n-- && fast != NULL){
fast = fast->next;
}
// fast 再前进一步,以使得 slow 指针最终能够指向要删除节点的前一个节点
fast = fast->next;
// 同时移动 fast 和 slow 指针,直到 fast 指向链表末尾
while(fast) {
fast = fast->next;
slow = slow->next;
}
// slow 指针现在指向了要删除节点的前一个节点
// 执行删除操作
ListNode* tmp = slow->next; // 要删除的节点
slow->next = slow->next->next; // 将要删除的节点从链表中断开
// dummyHead 的 next 指针指向链表的头节点
ListNode *tmp1 = dummyHead->next;
// 释放要删除的节点和哑结点所占用的内存
delete tmp; // 删除要移除的节点
delete dummyHead; // 删除哑结点
// 返回链表的新头节点,即 tmp1
return tmp1;
}
};