这是我自己写的但是遍历了两边,时间复杂度为0(n)
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
// 创建虚拟头节点,可以更简洁地处理删除头节点的情况,如果不用虚拟指针还要考虑如果要删除的是链表的第一个节点(也就是头节点),你的代码会失败。
//因为如果 size - n == 0,那么 cur 就会指向 nullptr,然后访问 cur->next 会发生空指针解引用错误。
ListNode* dummyhead = new ListNode();
dummyhead->next = head;
// 计算链表的长度
ListNode* temp = head;
int size = 0;
while (temp != nullptr) {
temp = temp->next;
size++;
}
// 找到倒数第n个节点的前一个节点
ListNode* cur = dummyhead;
for (int i = 0; i < size - n; i++) {
cur = cur->next;
}
// 删除倒数第n个节点
cur->next = cur->next->next;
// 返回新的链表头(虚拟头节点的下一个节点)
return dummyhead->next;
}
};
如果用快慢指针的话,会更快一点
也是我自己写的
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
//首先创建虚拟结点,这样就不用考虑头节点是不是为空
ListNode* dummyhead=new ListNode();
dummyhead->next=head;
ListNode* fast=head;
ListNode* slow=dummyhead;
while(n--&&fast!=nullptr){
fast=fast->next;
}
while(fast!=nullptr){
fast=fast->next;
slow=slow->next;
}
slow->next=slow->next->next;
return dummyhead->next;
}
};