LeetCode 19. Remove Nth Node From End of List (刪除連結串列的倒數第 N 個結點)

題目

連結

https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list/

問題描述

給你一個連結串列，刪除連結串列的倒數第 n 個結點，並且返回連結串列的頭結點。

示例

輸入：head = [1,2,3,4,5], n = 2
輸出：[1,2,3,5]

提示

連結串列中結點的數目為 sz
1 <= sz <= 30
0 <= Node.val <= 100
1 <= n <= sz

思路

快慢指標，同時slow初始值為pre的值，確保刪除的結點剛好是倒數第n個點。

複雜度分析

時間複雜度 O(L)
空間複雜度 O(1)

程式碼

Java

    public ListNode removeNthFromEnd(ListNode head, int n) {
        ListNode pre = new ListNode();
        pre.next = head;
        ListNode fast = head;
        ListNode slow = pre;

        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 pre.next;
    }