【算法】删除链表的倒数第N个结点

【算法】删除链表的倒数第N个结点

出处

剑指offer 021

题解地址可参考：https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list/solution/shan-chu-lian-biao-de-dao-shu-di-nge-jie-dian-b-61/

题目

给你一个链表，删除链表的倒数第 n 个结点，并且返回链表的头结点。

示例 1：

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

示例 2：

输入：head = [1], n = 1
输出：[]

示例 3：

输入：head = [1,2], n = 1
输出：[1]

提示：

链表中结点的数目为 sz
1 <= sz <= 30
0 <= Node.val <= 100
1 <= n <= sz

进阶：你能尝试使用一趟扫描实现吗？

思路

快慢指针法：

由于我们需要找到倒数第 n 个节点，因此我们可以使用两个指针 first 和 second 同时对链表进行遍历，并且 first 比 second 超前 n 个节点。当 first 遍历到链表的末尾时，second 就恰好处于倒数第 n 个节点。

具体地，初始时 first 和 second 均指向头节点。我们首先使用 first 对链表进行遍历，遍历的次数为 n。此时，first 和 second 之间间隔了 n−1 个节点，即 first 比 second 超前了 n 个节点。

在这之后，我们同时使用 first 和 second 对链表进行遍历。当 first 遍历到链表的末尾（即 first 为空指针）时，second 恰好指向倒数第 n 个节点。

代码

public class M {

    public ListNode removeNthFromEnd(ListNode head, int n) {
        // 构造一个哑节点，next指向head
        ListNode dummy = new ListNode(0, head);
        // 快指针指向head节点
        ListNode first = head;
        // 慢指针指向哑节点
        ListNode second = dummy;
        // 快指针向后移动n个节点
        for (int i = 0; i < n; ++i) {
            first = first.next;
        }
        // 快指针和慢指针同时向后移动，直到快指针指向null为止
        while (first != null) {
            first = first.next;
            second = second.next;
        }
        // 慢指针向后移动两个节点，
        second.next = second.next.next;
        // 因为dummy和second是指向同一个节点的，所以返回哑节点的后一个节点即可
        ListNode ans = dummy.next;
        return ans;
    }
}

class ListNode {
    int val;
    ListNode next;

    ListNode(int val, ListNode next) {
        this.val = val;
        this.next = next;
    }
}