LeetCode刷题记录 NO.19删除链表的倒数第N个节点
题目描述
给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
说明:
给定的 n 保证是有效的。
进阶:
你能尝试使用一趟扫描实现吗?
来源:力扣
解题思路
最直接的思路,遍历链表,获取长度L,再遍历到第L-N+1个节点,将其删除,需要遍历两遍。
如果只遍历一遍,可以想到递归,在递归返回阶段,从后往前出栈,可以很方便地找到倒数第N个节点。
上代码
第一遍写的时候,很直接想到递归返回到倒数n+1个节点,将它的next指针指向倒数第n-1个节点从而删除倒数n节点。
/*class Solution {
public int i = 0;
public ListNode removeNthFromEnd(ListNode head, int n) {
if(head==null) return null;
head.next = removeNthFromEnd(head.next,n);
i++;
if(i==n+1) head.next=head.next.next;
return head;
}
}*/
但是有个致命的问题,就是链表只有1个节点的话,没有倒数第n+1和n-1个节点,懵逼了。
看了评论区的作业之后回过神来,这种操作太蠢了。
class Solution {
public int i = 0;
public ListNode removeNthFromEnd(ListNode head, int n) {
if(head==null) return null;
head.next = removeNthFromEnd(head.next,n);
i++;
if(i==n) return head.next;
return head;
}
}
直接在找到倒数第n个节点之后返回next节点。