删除链表的倒数第N个结点
题目
删除链表的倒数第N个结点
给你一个链表,删除链表的倒数第 n
个结点,并且返回链表的头结点。
示例:
输入:head = [1,2,3,4,5], n = 2
输出:[1,2,3,5]
力扣原题:https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list/
解法1-哈希表
思路:先遍历一次链表,将各个结点按顺序存入哈希表中。然后再删除指定结点。
C++代码实现:
/**
* 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* cru = head;
unordered_map<int, ListNode*> dic;
int i = 0;
while(cru != NULL){
dic.insert({i, cru});
i++;
cru = cru->next;
}
if(dic[i-n]==head){
head = head->next;
}else if(dic[i-n]->next == NULL){
dic[i-n-1]->next = NULL;
}else{
dic[i-n-1]->next = dic[i-n+1];
}
return head;
}
};
时间复杂度:O(n)
空间复杂度:O(n)
解法2-双指针(快慢指针)
思路:快慢指针,快指针先走n步,然后慢指针从头开始一起走。当快指针走到最后时,慢指针指向的就是倒数第n个结点。
算法流程:
- 特判:若头结点为空,或该链表只有1个结点,那么直接返回空指针。
- 快指针先走n步。
- 特判:若快指针为空,那么表示要删除的结点是头结点,直接返回
head->next
。 - 快慢指针一起走,当快指针的
next
指针为空时跳出循环。 - 此时慢指针的
next
指针指向的就是要删除的结点,将该结点删除。 - 返回链表头结点。
C++代码实现:
/**
* 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) {
// 特判:若头结点为空,或该链表只有1个结点,那么直接返回空指针
if(head == NULL || head->next == NULL) return NULL;
// fast:快指针
ListNode* fast = head;
// slow:慢指针
ListNode* slow = head;
// 快指针先走n步
for(int i=0;i<n;i++){
fast = fast->next;
}
// 特判:若快指针为空,那么表示要删除的结点是头结点,直接返回head->next
if(fast == NULL) return head->next;
// 快慢指针一起走,当快指针的next指针为空时跳出循环
while(fast->next != NULL){
slow = slow->next;
fast = fast->next;
}
// 此时慢指针的next指针指向的就是要删除的结点,将该结点删除
slow->next = slow->next->next;
// 返回链表头结点
return head;
}
};
时间复杂度:O(n)
空间复杂度:O(1)