题目:给定单链表的头指针和一个结点指针,定义一个函数在O(1)时间删除该结点。链表结点与函数定义如下:
struct ListNode
{
int value;
ListNode *next;
};
void Delete(ListNode *head,ListNode *deleted);
解答:正常的删除结点(结点p)都是知道被删除结点的前一个结点(q)q->next=p->next;delete p;但是这样需要从头开始遍历链表,时间复杂度就变成了O(N),题目要求是O(1),所以不能简单的这么做。可以换种思维考虑,要删除结点p,其实就是p->value删除,p->next只是一个指针,可以把p->value换成p->next->value;再删除p->next。这样就可以了。
void Delete(ListNode *head,ListNode *deleted)
{
ListNode *p;
if(head==NULL||deleted==NULL)
return;
if(deleted->next!=NULL) //删除的节点不是尾结点
{
p=deleted->next;
deleted->value=p->value;
deleted->next=p->next;
delete p;
}
else if(head->next==NULL) //链表只有一个结点为删除的结点
{
delete deleted;
head=NULL;
}
else //删除的结点为尾结点,此时需要从头遍历链表,找到被删除结点的前一个结点
{
p=head;
while(p->next!=deleted)
p=p->next;
p->next=NULL;
delete deleted;
}
}