删除链表的倒数第N个节点(Remove Nth Node From End of List)java_leetcode19
版权©: https://blog.youkuaiyun.com/zhangjingao/article/details/96988414
题干
给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
说明:
给定的 n 保证是有效的。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list
题解
我觉得使用笨方法遍历两遍是肯定可以的,可以采用一遍记住链表长度,第二遍计算下标删除。
本题要求一趟扫描,那么就需要一些空间上的支持,这里我才用集合,对链表进行遍历一遍,将所有节点存进list,那么在结束后,我就可以计算出要删除的节点,删除掉返回即可。
代码
public ListNode removeNthFromEnd(ListNode head, int n) {
if (head == null || (head.next == null && n == 1)) {
return null;
}
List<ListNode> listNodes = new ArrayList<>();
int index = 1;
while (head != null) {
listNodes.add(head);
head = head.next;
index++;
}
int cur = index - 1 - n - 1;
ListNode tmp;
if (cur == -1) {
return listNodes.get(0).next;
} else {
tmp = listNodes.get(cur);
}
if (n == 1) {
tmp.next = null;
} else {
tmp.next = tmp.next.next;
}
return listNodes.get(0);
}