这道题题目可以描述为:给定单向链表的头指针和一个结点指针,定义一个函数在O(1)时间删除该结点。
本文中有关链表的结构体以及接口定义如下:
typedef int DataType;
typedef struct ListNode{
DataType data;
struct ListNode *next;
} ListNode;
//初始化
void ListInit(ListNode **ppfirst)
{
assert(ppfirst != NULL);
*ppfirst = NULL;
}
这道题的思路分为三步:
(1)如果要删除结点i,先把i的下一个结点j的内容复制到i,然后把i的指针指向结点j的下一个结点,此时再删除结点j,其效果刚好是把结点i删除了(如图);
(2)对于尾结点,从头结点开始遍历找到为结点的前一个结点,然后完成删除操作;如果链表中只有一个结点,即删除头结点,删除之后把链表的头结点置为NULL;
(3)对于n-1个非尾结点,可以在O(1)时把这个结点删除;对于尾结点,由于要遍历查找,它的时间复杂度为O(n);因此总的平均时间复杂度为[(n-1)*O(1)+O(n)]/n,结果还是O(1);
代码如下:
void DeleteNode(ListNode **phead, ListNode *pdelete)
{
assert(phead && pdelete);
//要删除的不是尾结点
if (pdelete->next != NULL)
{
ListNode *pnext = pdelete->next;
pdelete->data = pnext->data;
pdelete->next = pnext->next;
free(pnext);
pnext = NULL;
}
//删除头结点
else if (*phead == pdelete)
{
free(pdelete);
pdelete = NULL;
*phead = NULL;
}
//删除尾结点
else
{
ListNode * pNode =* phead;
while (pNode->next != pdelete)
{
pNode = pNode->next;
}
pNode->next = NULL;
free(pdelete);
pdelete = NULL;
}
}