题目

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

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

题解

public ListNode removeNthFromEnd(ListNode head, int n) {
    int count = 0;
    // 节点缓存
    Map<Integer, ListNode> cache = new HashMap<>(8);
    ListNode cursor = head;
    while (Objects.nonNull(cursor)) {
        cache.put(++count, cursor);
        cursor = cursor.next;
    }

    // 移除节点索引
    int index = count - n + 1;
    // 移除头结点 若只有一个节点 头尾节点是同一个
    if (index == 1) {
        head = count == 1 ? null : head.next;
    } else if (index == count) {
        cache.get(index - 1).next = null;
    } else {
        // 移除中间节点
        cache.get(index - 1).next = cache.get(index + 1);
    }

    return head;
}