一、题目
在O(1)时间删除链表结点:给定单向链表的头指针和一个结点指针,定义一个函数在O(1)时间删除该结点。
struct ListNode
{
int m_nValue;
ListNode* m_pNext;
};
void DeleteNode(ListNode** pListHead, ListNode* pToBeDeleted)
二、解法
分析:在单向链表中删除一个节点,常规的做法无疑是从链表的头结点开始,顺序遍历查找要删除的节点,并在链表中删除该节点。这种思路由于需要顺序查找,时间复杂度自然就是O(n)。
那是不是一定需要得到被删除的节点的前一个节点呢?答案是否定的。我们可以很方便的得到要删除的节点的下一个节点。如果我们把下一个节点的内容复制到需要删除的节点上覆盖原有的内容,再把下一个节点删除,那是不是就相当于把当前需要删除的节点删除了。该思路存在一个问题:如果要删除的节点位于链表的尾部,那么它就没有下一个节点,怎么办?我们仍然从链表的头结点开始顺序遍历得到该节点的前序节点,并完成删除操作。
最后需要注意的是,如果链表只有一个节点,而我们又要删除链表的头结点,那么此时我们在删除节点之后,还需要把链表的头结点设置为nullptr。
void DeleteNode(ListNode** pListHead, ListNode* pToBeDeleted)
{
if(!pListHead || !pToBeDeleted)
return;
// 要删除的结点不是尾结点
if(pToBeDeleted->m_pNext != nullptr)
{
ListNode* pNext = pToBeDeleted->m_pNext;
pToBeDeleted->m_nValue = pNext->m_nValue;
pToBeDeleted->m_pNext = pNext->m_pNext;
delete pNext;
pNext = nullptr;
}
// 链表只有一个结点,删除头结点(也是尾结点)
else if(*pListHead == pToBeDeleted)
{
delete pToBeDeleted;
pToBeDeleted = nullptr;
*pListHead = nullptr;
}
// 链表中有多个结点,删除尾结点
else
{
ListNode* pNode = *pListHead;
while(pNode->m_pNext != pToBeDeleted)
{
pNode = pNode->m_pNext;
}
pNode->m_pNext = nullptr;
delete pToBeDeleted;
pToBeDeleted = nullptr;
}
}