19. 删除链表的倒数第N个节点

2019/01/15 11:25 上午 posted in  leetcode

问题

给定一个链表,删除链表的倒数第 \(n\) 个节点,并且返回链表的头结点。

示例:

给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.

说明:

给定的 \(n\) 保证是有效的。

进阶:

你能尝试使用一趟扫描实现吗?

解法

使用双指针来处理。\(fast\)指针先走\(n\)步,然后\(fast\)指针与\(slow\)指针同时推进,当\(fast\)指针走到链表尾时,\(slow\)指针正好走到倒数\(n+1\)的位置。这时候只需要\(slow.next = slow.next.next\)即可,但是如果直接使用\(head\)作为开头有很多的不便。比如\(n\)等于链表长度,此时,要删除的其实为\(head\)节点,但是此时很难去判断究竟是要删除\(slow\)节点,还是\(slow.next\)节点。
为了排除这种情况,我们引入一个\(dummy\)节点,\(dummy.next = head\),在删除时永远只删除\(slow.next\),这样我们无论怎么处理链表,最终链表的\(head\)均为\(dummy.next\)。

代码

java实现

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode removeNthFromEnd(ListNode head, int n) {
        ListNode dummy = new ListNode(0);
        dummy.next = head;
        ListNode fast = dummy;
        ListNode slow = dummy;
        //fast节点先走n步,此处不需要考虑fast.next是否为null,题目保证了n是个有效的数字。
        for(int i = 1; i <= n; i++) {
            fast = fast.next;
        }
        //开始同时推进fast与slow,这里循环跳出条件是fast.next ==null,即fast已经到了end。
        while (fast.next != null) {
            fast = fast.next;
            slow = slow.next;
        }
        //此时slow已经到了倒数n+1的位置。但是仍然需要判断slow.next是否为null,防止NPE
        if (slow.next != null) {
            slow.next = slow.next.next;
        }
        //dummy.next为head。返回即可
        return dummy.next;
        
    }
}