一、题目描述
给你一个链表,删除链表的倒数第 n 个结点,并且返回链表的头结点。
输入:head = [1,2,3,4,5], n = 2
输出:[1,2,3,5]
二、思路
用虚拟头结点+双指针的方法解决该题,我们知道题目要求我们返回的是链表的头结点,所以头结点我们需要用一个变量指向它,保存它的位置,所以虚拟头结点完美胜任该功能,再初始化两个指向虚拟头结点的指针,使两个指针之间的跨度是n(也就是fast指针先走n位),然后在快指针不越界的条件下两个指针同时移动,保持跨度的作用就是当快指针指向末尾null的时候,慢指针指向刚刚好就是我们要删除的元素位置,但是我们希望慢指针指向的是要删除元素的前一个位置,所以两个指针之间的跨度应该是n+1,就是让快指针多走一步(或者用快指针的next指针判断是否为null),慢指针满足了我们的要求,就可以执行删除操作了,也就是慢指针的next指向下下个元素,最后返回头结点。
综上所述,分四步:
- 初始化一个指向头结点的虚拟节点
- 让快指针先走n步,使快慢指针跨度是n
- 判断快指针下一位是否是null,true就说明慢指针处于删除元素的前一个位置了,false就继续移动快慢指针
- 删除指定位置元素,返回头结点
三、代码展示
var removeNthFromEnd = function(head, n) {
let ret = new ListNode(0, head), // 定义一个虚拟头结点指向头结点
slow = fast = ret; // 快慢指针指向虚拟头结点
while(n--) fast = fast.next; // 快指针移动n-1步
while (fast.next !== null) { // 若改为fast,慢指针就指向了倒数第n位,需要指向它的n-1位
fast = fast.next;
slow = slow.next
};
slow.next = slow.next.next; // 此时慢指针在倒数n-1位,可以删除倒数第n个节点
return ret.next; // ret是虚拟头结点,指向头结点
};