您的位置:首页 > Web前端

【剑指 offer】(十三)—— 在 O(1) 时间删除链表结点

2016-04-09 11:58 330 查看
struct ListNode
{
int val;
ListNode* next;
};
void DeleteNode(ListNode** pListHead, ListNode* pToBeDeleted);


一般情况下,我们需要从头开始查找(时间复杂度为 O(n)),找到被删除节点的前一个结点。正如题目所言,是否一定需要得到被删除节点的前一个结点呢?答案是否定。我们可以找到一种等价做法,如果我们把下一个节点的内容(无非是 val 和指向下一个元素的指针)覆盖自身的内容,再把下一个结点删除。

这里需要注意两个问题,如果被删除的节点位于链表的尾部,那么他就没有下一个结点(next == NULL),此时值得从节点开始遍历,找到倒数第二个节点。

如果链表只有一个节点,而我们又要删除链表的头结点(也是尾节点),需要将链表的头结点置为 NULL。

void DeleteNode(ListNode** pListHead, ListNode* pToBeDeleted)
{
if (!pListHead || !pToBeDeleted)
return;

if (pToBeDeleted->next != NULL)
{
ListNode* pNext = pToBeDeleted->next;
pToBeDeleted->val = pNext->val;
pToBeDeleted->next = pNext->next;
delete pNext;
pNext = NULL;
}
else if (*pListHead == pToBeDeleted)
{
delete pToBeDeleted;
pToBeDeleted = NULL;
*pListHead = NULL;
}
else
{
ListNode* pNode = *pListHead;
while (pNode->next != pToBeDeleted)
pNode = pNode->next;
pNode->next = NULL;
delete pToBeDeleted;
pToBeDeleted = NULL;
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: