[LeetCode] Remove Nth Node From End of List

目录

问题

分析

代码

---

问题

    给定一个链表，删除链表的倒数第 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;
    }