给你一个链表,删除链表的倒数第 n 个结点,并且返回链表的头结点。
进阶:你能尝试使用一趟扫描实现吗?
这个题目的难度在于。测试样例
输入: [1]
输出:[]
使用快慢指针
public ListNode removeNthFromEnd(ListNode head, int n) {
if(head==null){
return null;
}
ListNode node1=head;
ListNode node2=head;
while(n>0){
node1=node1.next;
n--;
}
while( node1!=null && node1.next!=null){
node1=node1.next;
node2=node2.next;
}
node2.next=node2.next.next;
return head;
}
这里存在问题,当只有一个节点,并且要删除头节点的时候,会报错,不满足条件,
因为node2.next 是null。
解决方案:
增加虚拟的头节点,也使用快慢指针
ListNode pre = new ListNode(0);
pre.next = head;
ListNode start = pre, end = pre;
while(n != 0) {
start = start.next;
n--;
}
while(start.next != null) {
start = start.next;
end = end.next;
}
end.next = end.next.next;
return pre.next;