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

来源：互联网发布：程序员上班3天被劝退编辑：程序博客网时间：2024/05/29 04:03

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;    }}

0 0