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

来源:互联网 发布:淘宝樱花牌油画棒 编辑:程序博客网 时间:2024/06/10 22:35

题目:给定单向链表的头指针和一个节点指针,定义一个函数在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上:点我前往

0 0
原创粉丝点击