题目链接:https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list/
题目描述
给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
说明:
给定的 n 保证是有效的。
进阶:
你能尝试使用一趟扫描实现吗?
思路
一趟遍历
设置快慢指针,快指针先走n步后,快慢指针同时走;
当快指针到达链表尾部(nullptr)时,慢指针刚好到达倒数第n个节点;
代码
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
if(!head || n <= 0) return head;
ListNode *headNode = new ListNode(-1);
ListNode *fast = head; // 快指针
ListNode *slow = head; // 慢指针
ListNode *prev = nullptr; // 慢指针前一个节点
for (int i = 0; fast && i < n; ++i)
fast = fast->next; // 快指针先走n步
if(!fast) return head->next; // 删除的是头结点
while (fast){ // 当快指针走到空,则慢指针指向导师第k个节点
prev = slow;
slow = slow->next;
fast = fast->next;
}
prev->next = slow->next; // 删除节点
return head;
}
};