题目:在给定单向链表的头指针和一个结点指针,定义一个函数在O(1)时间内删除该节点。链表的结点定义如下:
struct ListNode {
int val;
ListNode *next;
};
要分析几种情况:
1.空链表
2.只有一个结点
3.删除的结点是尾结点
void DeleteNode(ListNode *pNode,ListNode *pDeleted) {
if(!pNode || !pDeleted) return;
//要删除的节点不是尾结点
if(pDeleted->next != NULL) {
ListNode *pNext = pDeleted->next;
pDeleted->value = pNext->value;
pDeleted->next = pNext ->next;
delete pNext;
pNext = NULL;
}
//链表只有一个节点,删除头结点(也是尾节点)
else if(pNode == pDeleted) {
free(pDeleted);
pDeleted = NULL;
pListNode= NULL;
}
//链表中有多个节点,删除尾结点
else {
ListNode *p = pNode;
while(p ->next != pDeleted) {
p = p->next;
}
p->next =NULL;
delete pDeleted;
pDeletd = NULL;
}
}
}
总结:对于n-1个非尾结点而言,我们可以在O(1)时把下一个节点的内存复制覆盖要删除的结点,并删除下一个节点;对于尾结点而言,由于仍要顺序查找到尾结点的前面一个节点,时间复杂度是O(n)。因此总的平均时间复杂度是[(n-1)*O(1) +O(n)]/n 结果还是O(1)