NC53 删除链表的倒数第n个节点

题目链接：https://www.nowcoder.com/practice/f95dcdafbde44b22a6d741baf71653f6

题目描述

给定一个链表，删除链表的倒数第 n 个节点并返回链表的头指针
例如，给出的链表为: 1→2→3→4→5, n= 2.
删除了链表的倒数第 n 个节点之后,链表变为1→2→3→5.

数据范围： 链表长度 0≤n≤1000，链表中任意节点的值满足 0≤val ≤100

要求：空间复杂度 O(1)，时间复杂度 O(n)
备注：题目保证 n 一定是有效的

示例 1：

输入：{1,2},2
返回值：{2}

解题代码

import java.util.*;

/*
 * public class ListNode {
 *   int val;
 *   ListNode next = null;
 * }
 */

public class Solution {
    /**
     * 
     * @param head ListNode类 
     * @param n int整型 
     * @return ListNode类
     */
    public ListNode removeNthFromEnd (ListNode head, int n) {
        // write code here
        if (head == null || n == 0) return head;
        ListNode cur = head, pre = head;
        for (int i = 0; i < n; i++) {
            if (cur == null) return head;
            cur = cur.next;
        }
        if (cur == null) {
            return head.next;
        }
        while (cur.next != null) {
            cur = cur.next;
            pre = pre.next;
        }
        pre.next = pre.next.next;
        return head;
    }
}