1、题目:给定单向链表的头指针和一个结点指针,定义一个函数在O(1)时间删除该结点。
struct ListNode
{
int m_nValue;
ListNode* m_pNext;
};
解法:最直观的做法是从链表的头结点开始,顺序遍历查找要删除的结点,时间复杂度为O(n)。
之所以需要从头开始查找,是因为我们需要得到将被删除的结点的前面一个结点。单向链表中,结点中没有指向前一个结点的指针。
1)如果我们把下一个结点的内容复制到需要删除的结点上覆盖原有的内容,再把下一个结点删除,是不是就相当于把当前需要删除的结点删除了?
即要删除结点i,先把i的下一个结点j的内容复制到i,然后把i的指针指向结点j的下一个结点。此时再删除结点j,其效果刚好是把结点i给删除了。
2)如果要删除的结点位于链表的尾部,那么它就没有下一个结点,怎么办?
我们仍然从链表的头结点开始,顺序遍历得到该结点的前序结点,并完成删除。
3)若链表中只有一个结点,而我们又要删除链表的头结点(也是尾结点),此时再删除结点之后,还需要把链表的头结点置为NULL。
void DeleteNode(ListNode** pListHead, ListNode* pToBeDeleted)
{
if(!pListHead || !pToBeDeleted)
return;
//要删除的结点不是尾结点
if(pToBeDeleted->m_pNext != NULL)
{
ListNode* pNext = pToBeDeleted->m_pNext;
pToBeDeleted->m_nValue = pNext->m_nValue;
pToBeDeleted->m_pNext = pNext->m_pNext;
delete pNext;
pNext = NULL;
}
//链表只有一个结点,删除头结点(也是尾结点)
else if(*pListHead == pToBeDeleted)
{
delete pToBeDeleted;
pToBeDeleted = NULL;
*pListHead = NULL;
}
//链表中有多个结点,删除尾结点
else
{
ListNode* pNode = *pListHead;
while(pNode->m_pNext != pToBeDeleted)
{
pNode = pNode->m_pNext;
}
pNode->m_pNext = NULL;
delete pToBeDeleted;
pToBeDeleted = NULL;
}
}
对于n-1个非尾结点而言,我们可以在O(1)时把下一个结点的内存复制覆盖要删除的结点,并删除下一个结点。对于尾结点而言,由于仍然需要顺序查找,时间复杂度是O(n)。因此总的平均时间复杂度是[(n-1)*O(1)+O(n)]/n,结果还是O(1)。