给你一个链表,删除链表的倒数第 n
个结点,并且返回链表的头结点。
计算链表长度:
首先从头结点开始对链表进行一次遍历,得到链表的长度L。然后从头结点对链表再进行一次遍历,当遍历到L-n+1个结点,它就是要删除的结点
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode* fast=head;
int length=0;
while(fast!=nullptr){
++length;
fast=fast->next;
}
int site=length-n-1;
if(site<0){
head=head->next;
return head;
}
fast=head;
while(site!=0){
fast=fast->next;
--site;
}
fast->next=fast->next->next;
return head;
}
};
代码中没有回收内存。有些小缺憾,但是懒得加了。注意一下。
栈的思想:
在遍历链表的同时,将所有节点入栈,根据栈先进后出的特点,弹出栈第n个节点就是需要删除的结点,目前栈顶的结点就是待删除节点的前驱结点。
官方代码
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode* dummy = new ListNode(0, head);
stack<ListNode*> stk;
ListNode* cur = dummy;
while (cur) {
stk.push(cur);
cur = cur->next;
}
for (int i = 0; i < n; ++i) {
stk.pop();
}
ListNode* prev = stk.top();
prev->next = prev->next->next;
ListNode* ans = dummy->next;
delete dummy;
return ans;
}
};
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list/solution/shan-chu-lian-biao-de-dao-shu-di-nge-jie-dian-b-61/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
自己实现:
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode* fast=head;
ListNode* wula=new ListNode(0,head);
stack<ListNode*> stk;
stk.push(wula);
while(fast!=nullptr){
stk.push(fast);
fast=fast->next;
}
while(n!=0){
stk.pop();
--n;
}
ListNode* prev=stk.top();
if(prev==wula){
head=head->next;
return head;
}
prev->next=prev->next->next;
return head;
}
};
双指针的思想:
使用两个指针first和second同时对链表进行遍历,并且first比second超前n个结点,当first遍历到链表末尾,second恰好处于倒数第n个节点。
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode* dummy=new ListNode(0,head);
ListNode* fast=dummy;
ListNode* slow=dummy;
while(n!=0){
fast=fast->next;
--n;
}
while(fast->next!=nullptr){
slow=slow->next;
fast=fast->next;
}
slow->next=slow->next->next;
return dummy->next;
}
};
注意如果链表长1,如果返回head,返回的是未修改的链表,因此,要返回dummy->next