目录
题目描述
方法一:计算链表的长度
先从头到尾遍历链表,得到链表的长度count,结点编号从1开始起算的话,第(count - n + 1)个结点就是要删除的结点。leetcode官方题解,给出的办法是再遍历一次,遍历到第count-n+1个结点就可以做删除操作了。我用的是用空间换时间,第一次遍历的时候用哈希表把所有结点的地址存起来,哈希键就是结点的序号,这样就不用第二次遍历链表,但同时增加了空间开销。
遍历两次:
/**
* 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) {
ListNode* dummy = new ListNode(0,head);
int length = 0;
while(head){
++length;
head = head->next;
}
ListNode* cur = dummy;
for(int i=0;i < length-n;i++){
cur = cur->next;
}
cur->next = cur->next->next;
ListNode* ans = dummy->next;
delete dummy;
return ans;
}
};
题目说结点的数目是1到30个,所以哈希表ListNode* nodes[31]。但是下面的代码是将结点序号从0开始起算的,所以实际上nodes[30]就够了。由于结点的序号从0开始起算,所以要删除的结点就是第(count - 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) {
ListNode* nodes[31];
ListNode* cur = head;
int count = 0;
while(cur){
nodes[count++] = cur;
cur = cur->next;
}
int todelete = count - n;
if(todelete == 0){
return nodes[todelete]->next;
}else{
nodes[todelete-1]->next = nodes[todelete]->next;
}
return head;
}
};
方法二:栈
遍历链表的时候将所有结点入栈,遍历结束后再从栈弹出n个结点,弹出的第n个结点就是要删除的结点。弹出n个结点之后,栈顶结点就是要删除结点的前驱pre,将pre->next置为pre->next->next即可解决问题。
这种方法巧妙地利用了栈的特性找倒数第n个结点。时间开销上,相当于遍历了两次链表。空间开销也是O(L)。
/**
* 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) {
ListNode* dummy = new ListNode(0,head);
stack<ListNode*> node_stack;
ListNode* cur = dummy;
while(cur){
node_stack.push(cur);
cur = cur->next;
}
for(int i = 0;i < n;i++){
node_stack.pop();
}
ListNode* pre = node_stack.top();
pre->next = pre->next->next;
ListNode* ans = dummy->next;
delete dummy;
return ans;
}
};
方法三:双指针
使用两个指针遍历链表,快指针fast比慢指针slow超前n个结点。当fast到达链表末尾(为空)时,slow就是倒数第n个结点。只需要遍历一次链表,这种方法才是本题真正要考察的方法,可以和第2095题对比学习2095. 删除链表的中间节点
/**
* 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) {
ListNode* dummy = new ListNode(0,head);
ListNode* fast = head;
ListNode* slow = head;
ListNode* pre = dummy;
//fast比slow超前n个节点,fast到达链表末尾时,slow就是倒数第n个结点
while(n){
fast = fast->next;//题目自己保证右边的fast不会为空
n--;
}
while(fast){
fast = fast->next;
pre = slow;
slow = slow->next;
}
pre->next = slow->next;
delete slow;
ListNode* ans = dummy->next;
delete dummy;
return ans;
}
};