面试题13：在O(1)时间 内删除链表节点

题目：给定单向链表的头指针和一个节点指针，定义一个函数在O(1)时间删除该节点。链表节点与函数的定义如下：

struct ListNode
{
    int value;
    ListNode* next;
}

void DeleteNode(ListNode* head,ListNode* pToBeDeleted);

---

分析

在单向链表中删除一个节点，通常的做法是从链表头开始遍历，如果某个节点的next等于要删除的节点，就将该next改为待删除节点的下一个节点，然后删除该链表节点即可。

代码如下：

void DeleteNode(ListNode* head,ListNode* pToBeDeleted)
{
    if(head==NULL || pToBeDeleted==NULL)
        return;
    ListNode* cur=head;
    if(cur==pToBeDeleted)
    {
        head=cur->next;
        delete pToBeDeleted;
        return head;
    }
    while(cur!=NULL)
    {
        if(cur->next==pToBeDeleted)
        {
            cur->next==pToBeDeleted->next;
            delete pToBeDeleted;
            return;
        }
        cur=cur->next;
    }
}

由于这样的思路需要顺序查找，时间复杂度自然就是O(n)了。

因为我们要删除某个节点，如果不是第一个节点的话，就需要更改它前面节点的next值。而单链表是无法逆行的，所以我们可以按如下步骤做：

1. 1. 如果待删除节点不是最后一个节点，就用他的next节点的value覆盖它的value，然后删掉它的next节点。
2. 2. 如果是最后一个节点，还是老老实实的顺序遍历吧。

But，这种方法会带来一个问题，就是可能会使原来指向待删除节点的next节点的指针在删除操作之后失效！！任何事物都是有两面性的，将删除操作时间复杂度降到了O(1)，就要接受这种类似于迭代器失效问题的后果。

代码如下：

void DeleteNode(ListNode** pListHead,ListNode* pToBeDeleted)
{
    if(pListHead==NULL || *pListHead==NULL || pToBeDeleted==NULL)
        return;
    if(pToBeDeleted->next!=NULL)
    {
        ListNode* tmp=pToBeDeleted->next;
        pToBeDeleted->value=tmp->value;
        pToBeDeleted->next=tmp->next;
        delete tmp;
    }
    else
    {
        ListNode* cur=*pListHead;
        if(cur==pToBeDeleted)
        {
            *pListHead=cur->next;
            delete cur;
            return;
        }
        while(cur->next!=NULL)
        {
            if(cur->next==pToBeDeleted)
            {
                ListNode* tmp=cur->next;
                cur->next=tmp->next;
                delete tmp;
                return;
            }
            cur=cur->next;
        }
    }
}

以上

如果你有任何想法或是可以改进的地方，欢迎和我交流！

完整代码及测试用例在github上：点我前往