剑指 Offer II 021. 删除链表的倒数第 n 个结点
题目
给定一个链表,删除链表的倒数第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
进阶:能尝试使用一趟扫描实现吗?
注意:本题与主站19题相同
题解
public ListNode removeNthFromEnd(ListNode head, int n) {
// 缓存位置及节点
ListNode[] nodes = new ListNode[30];
ListNode cursor = head;
// 节点数量
int cnt = 0;
while (cursor != null) {
nodes[cnt++] = cursor;
cursor = cursor.next;
}
// 只有一个节点
if (cnt == 1) {
return null;
}
if (n == cnt) {
// 头结点
head = head.next;
} else if (n == 1) {
// 最后一个节点 将尾节点置空
nodes[cnt - 2].next = null;
} else {
// 中间节点
nodes[cnt - n - 1].next = nodes[cnt - n + 1];
}
return head;
}