中等
题目描述:
给你一个链表,删除链表的倒数第 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 <= 300 <= Node.val <= 1001 <= n <= sz
思路:
- 定义fast指针和slow指针,初始值为虚拟头结点,如图:
- fast首先走n + 1步 ,为什么是n+1呢,因为只有这样同时移动的时候slow才能指向删除节点的上一个节点(方便做删除操作),如图:
- fast和slow同时移动,直到fast指向末尾,如题:
以上是大致思路,但n没必要+1, 我只需要保证快慢指针不指向NULL就可以了,使用node->next != NULL做判断,也更好理解。
解题:
C语言版:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* struct ListNode *next;
* };
*/
struct ListNode* removeNthFromEnd(struct ListNode* head, int n) {
// 创建虚拟指针指向头指针
struct ListNode dumyNode = {0, head};
struct ListNode* dumyhead = &dumyNode;
// 定义快慢指针
struct ListNode* slow = dumyhead;
struct ListNode* fast = dumyhead;
// 先让快指针
while(n-- && fast->next!=NULL){
fast = fast->next;
}
// 快慢指针再同时移动,让slow指向待删除节点的前一个节点
while(fast->next!=NULL){
fast = fast->next;
slow = slow->next;
}
// 删除节点并释放内存
struct ListNode* delNode = slow->next;
slow->next = slow->next->next;
free(delNode);
// 最后返回头节点
return dumyhead->next;
}