给你一个链表,删除链表的倒数第 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
进阶:你能尝试使用一趟扫描实现吗?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
实现思路:
快慢指针。
实现代码:
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode *l=head,*m=head,*r=head;
for(int i=0;i<n;i++){
r=r->next;
}
if(r!=NULL){
r=r->next;
m=m->next;
}
while(r!=NULL){
r=r->next;
m=m->next;
l=l->next;
}
if(m==head) return m->next;
if(m!=NULL) l->next=l->next->next;
delete(m);
return head;
}
};
这是我自己写的思路。
不过在官方题解里我又学到了一点,创建一个哑结点,就可以回避要删除的结点是头结点的讨论。
修改后的代码:
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode *ya=new ListNode();
ya->next=head;
ListNode *l=ya,*m=head,*r=head;
for(int i=0;i<n;i++) r=r->next;
while(r!=NULL){
r=r->next;
l=l->next;
}
m=l->next;
l->next=m->next;
delete(m);
return ya->next;
}
};