第一种方法:略
方法一:计算链表长度思路与算法
一种容易想到的方法是,我们首先从头节点开始对链表进行一次遍历,得到链表的长度L。随后我们再从头节点开始对链表进行一次遍历,当遍历到第L一n+1个节点时,它就是我们需要删除的节点。
第二种方法:栈
我们也可以在遍历链表的同时将所有节点依次入栈。根据栈「先进后出」的原则,我们弹出栈的第n个节点就是需要删除的节点,并且目前栈顶的节点就是待删除节点的前驱节点。这样一来,删除操作就变得十分方便了。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
stack<ListNode*>head_stack;
if(head->next==nullptr){return nullptr;}
for(struct ListNode* p=head;p!=nullptr;p=p->next)
{
head_stack.push(p);
}
for(int i=1;i<n;i++)
{
if(!head_stack.empty())
head_stack.pop();
}
if(head_stack.size()==1)
{
head=head->next;
return head;
}
struct ListNode*p=head_stack.top();
head_stack.pop();
struct ListNode*ip=head_stack.top();
ip->next=p->next;
return head;
}
};
使用 了栈 空间复杂度为O(n).
方法3:使用快慢指针
方法三:双指针思路与算法
我们也可以在不预处理出链表的长度,以及使用常数空间的前提下解决本题。
由于我们需要找到倒数第n个节点,因此我们可以使用两个指针first和second同时对链表进行遍历,并且first 比 second超前n个节点。当first遍历到链表的末尾时,second就恰好处于倒数第n个节点。具体地,初始时 first和 second均指向头节点。我们首先使用first对链表进行遍历,遍历的次数为n。此时,first和second之间间隔了n-1个节点,即first 比 second超前了n个节点。
在这之后,我们同时使用first.和second对链表进行遍历。当first遍历到链表的末尾(即first为空指针)时,second恰好指向倒数第n个节点。
根据方法一和方法二,如果我们能够得到的是倒数第n个节点的前驱节点而不是倒数第n个节点的话,删除操作会更加方便。因此我们可以考虑在初始时将second指向哑节点,其余的操作步骤不变。这样一来,当first遍历到链表的末尾时,second的下一个节点就是我们需要删除的节点。
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode* dummy = new ListNode(0, head);
ListNode* first = head;
ListNode* second = dummy;
for (int i = 0; i < n; ++i) {
first = first->next;
}
while (first) {
first = first->next;
second = second->next;
}
second->next = second->next->next;
ListNode* ans = dummy->next;
delete dummy;
return ans;
}
};