算法-O(1)时间删除链表的指定结点

来源:互联网 发布:python java socket 编辑:程序博客网 时间:2024/05/17 08:47

题目:给定一个链表的头指针和一个结点的指针,定义一个函数在O(1)时间删除该结点。链表结点与函数的定义如下:

struct ListNode{  int value;  ListNode *next;};
void DeleteNode(ListNode** pListHead, ListNode* pToBeDeleted);

解题思路:
首先,乍一看这个名字而不看题目要求的话,这是功能是不可能在O(1)时间完成的,我们在之前讨论过关于单链表插入与删除的问题,链表的性质决定了不可能在O(1)下完成。所以题目给了我们其他的内容:要删除的结点的指针

那么按照之前单链表删除的思想来考虑的话,可以完成这件事,假设要删除的结点为i,i的前一个结点为h,那么我们需要找到h,并把h的next指针变为i的next指针,并释放i。

h->next = i->next;free(i);

这样的话i就删除了,不贴图了,前面的博客都有。但是这个思路不满足题目要求,因为单链表只有一个指向next的指针,那么我们想找到h只能从头遍历,一旦出现遍历这个东西,显然时间复杂度就不会是O(1)。

那么借助i前面的结点h是不行了,我们可以考虑能不能借助i后面的结点j,因为链表找到后面的结点只需要一步。我们只需要把j的value赋值给i的value,那么问题就变得清晰了:
要删除的结点变成了j,而i就是j前面的一个结点,而i是题目给的,不需要遍历得到。我们把问题从删除i找h,变换到了删除j找i上。

这里写图片描述

但是仅仅上面的内容还不够全面,因为我们利用了i的下一个结点j完成这个任务,但是如果i是尾结点呢
这时我们还需要借助i的前一个结点h这种方法,即让h的next为null,并释放i。
那么这样来看的话,又有了一个问题,i如果没有h也没有j呢?显然链表只有一个结点,而这个结点就是要删除的那个。

代码实现

void DeleteNode(ListNode** pListHead, ListNode* pToBeDeleted){    if(!pListHead || !pToBeDeleted)        return;    // 要删除的结点不是最后一个结点    if(pToBeDeleted->next != NULL)    {        ListNode* pNext = pToBeDeleted->next;        pToBeDeleted->value = pNext->value;        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;    }}

最后我们需要有两个注意的方法:
(1)这种方法有一个假设条件,那就是i确实出现在链表中。
(2)代码中还是出现了循环while,但是好在这个while在if里面,也就是遍历是有条件的,这个条件就是一个长度是n的链表,要删除的结点是尾结点,这个概率是1/n。除了这个之外,代码中其他部分的时间复杂度都是O(1),所以平均来看的话,整体的时间复杂度满足O(1)。

原创粉丝点击