leetcode题目19: 删除链表的倒数第N个节点（java）

题目描述

给定一个链表，删除链表的倒数第 n 个节点，并且返回链表的头结点。给定的 n 保证是有效的。

示例

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

代码

public ListNode removeNthFromEnd(ListNode head, int n) {
    if (head == null || n <= 0) {
        return head;
    }
    ListNode slow = head;
    ListNode fast = head;
    for (int i = 0; i < n - 1; i++) {
        if (fast == null) {
            return head;
        }
        fast = fast.next;
    }

    if (fast.next == null) {
        // 说明是删除第一个节点
        return slow.next;
    }

    fast = fast.next;
    while (fast.next != null) {
        fast = fast.next;
        slow = slow.next;
    }

    slow.next = slow.next.next;
    return head;
}