题目
Given a linked list, remove the n-th node from the end of list and return its head.
Example:
Given linked list: 1->2->3->4->5, and n = 2.
After removing the second node from the end, the linked list becomes 1->2->3->5.
分析
- 链表的增删改查都需要从头节点开始逐个往后移动,并且通常是移动到待处理节点的前一个节点。因此,定义一个指向头节点的虚拟头节点往往可以简化很多操作。
- 虚拟头节点定义好以后,再定义两个指针类型节点,通过这两个指针的移动可以解决问题
- 虚拟头节点指向真正的头节点,因此在对真正的头节点做处理的时候,不需要移动,就天然已经有待处理头节点的前一个节点了。省去了传统方式对头节点处理单独拎出来讨论的情况。
- 本题删除倒数第n个节点,可以先让快指针移动n个位置,再让慢指针开始移动,当快指针移动到末尾的时候,慢指针移动到倒数第n个节点前一位。
代码和注释
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* struct ListNode *next;
* };
*/
struct ListNode* removeNthFromEnd(struct ListNode* head, int n){
struct ListNode ret, *p, *q; //定义虚拟头节点和两个指针变量
ret.next = head; //虚拟头节点通过next属性指向真正的头节点
p = q = &ret; //这里初始把p,q都指向虚拟头节点
while(n--) p = p->next; //先让p指针移动n个位置
while(p->next) { // p节点继续移动,同时q开始移动,当p移动到末尾的时候q刚好移动到倒数第n个节点的前一个节点。
p = p->next;
q = q->next;
}
p = q->next; //重新利用p指针,把p指针指向的节点当做待删除节点
q->next = p->next;
free(p);
return ret.next; //这里不能return head,因为如果删除的恰好是第一个节点的时候,head已经被free掉。
}