目录
问题
给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
说明:
给定的 n 保证是有效的。
进阶:
你能尝试使用一趟扫描实现吗?
分析
要删除节点x,就是把这个节点的前驱的next改成 x.next,跳过x。所以要找到的是目标节点的前驱。
快慢指针是很容易想到的思路。两个指针从head开始,fast先走n步,然后两个指针同时走,直到fast到null,这时候slow在要删除的节点上。
对上面的思路稍加改动,新建一个节点preHead作为head的前驱,让slow从preHead开始走,fast还是从head开始,停下来时指向的就是目标节点的前驱,跳过slow的下一个节点就完成删除了。最后返回preHead.next。
为什么不能让fast多走一步,而是设置preHead来延长链表?
如果给一个长度n的节点,删除倒数第n个,就是第一个节点head,这个链表就不够fast走,会出现NullPointerException。另一方面,如果slow在初始位置就指向head,这时就找不到待删除节点的前驱,没法归约到上面的状态。诚然可以通过其他的修补解决问题,但那样显然不够简洁。用preHead挂在前面,可以把所有情况归结为返回preHead.next。
代码
public ListNode removeNthFromEnd(ListNode head, int n) {
ListNode preHead = new ListNode(0);
preHead.next = head;
ListNode fast = head, slow = preHead;
for(int i = 0; i < n; i++) {
fast = fast.next;
}
while(fast != null) {
fast = fast.next;
slow = slow.next;
}
slow.next = slow.next.next;
return preHead.next;
}