面试题18(一):在O(1)时间删除链表结点
题目:给定单向链表的头指针和一个结点指针,定义一个函数在O(1)时间删除该结点。
#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstring>
using namespace std;
struct ListNode{
int value;
ListNode* next;
};
/**
* 题目要求时间复杂度为O(1), 我们不能用遍历的思想对结点进行删除,
* 我们可以通过将要删除结点next的值赋值到要删除的结点,再让该结点指向next的next结点即可
* 特殊情况:1、要删除的结点为最后一个结点, 则需要从头开始遍历进行删除
* 2、链表只有一个结点,删除该结点后,需要将链表的头结点设置为NULL
* 对于尾节点而言, 由于仍然需要顺序查找, 时间复杂度是O(n), 因此,总的平均时间复杂度是
* [(n-1)xO(1)+O(n)]/n, 结果还是 O(1)
**/
void deleteNode(ListNode** pListHead, ListNode* pToDeleted) {
if(!pListHead || !pToDeleted) return ;
// 1、要删除的结点不是尾节点
if(pToDeleted->next!=NULL){
ListNode* pNext = pToDeleted->next;
pToDeleted->value=pNext->value;
pToDeleted->next=pNext->next;
delete pNext;
pNext=NULL;
}
// 2、链表只有一个结点
else if(*ListNode==pToDeleted){
delete pToDeleted;
pToDeleted=NULL;
*pListHead=NULL;
}
// 3、链表中有多个结点,删除尾节点
else {
ListNode* pNode = *pListHead;
while(pNode->next!=pToDeleted){
pNode=pNode->next;
}
pNode->next=NULL;
delete pToDeleted;
pToDeleted=NULL;
}
}
int main() {
return 0;
}