函数参数有两个:链表的头节点,需要删除的节点,两个都是ListNode*类型。
一. 常规思路,复杂度为O(n)
从头到尾遍历,找到该节点的前驱节点,用常规思路去设置指针。
即pre->next = deleted->next;delete deleted;
之所以要遍历,是因为是要找到被删除节点的前驱节点。
二. 高效解法:O(1)方法
可以用被删除节点的后继节点,去覆盖被删除节点,然后再删除后继节点,这样只需要对被删除节点、后继节点进行操作即可,无需遍历。
但是,如果被删除节点没有后继节点,即被删除节点是尾节点,则需要使用常规思路,去遍历找到前驱节点删除。最坏复杂度是O(n),平均复杂度是O(1)。
代码如下:
void DeleteNode(ListNode* head, ListNode* pDeleted)
{
if(head == NULL || pDeleted == NULL)
{
return;
}
if(pDeleted->next != NULL)//如果不是尾节点
{
ListNode* pNext = pDeleted->next;
pDeleted->val = pNext->val;
pDeleted->next = pNext->next;
delete pNext;
pNext = NULL;
}
else//如果是尾节点,只能从头到尾遍历,找到被删节点的前驱节点
{
ListNode* p = head;
ListNode* pre = NULL;
while(p != pDeleted)
{
pre = p;
p = p->next;
}
//pre=NULL时表示链表只有一个元素,并且被删除的就是该元素
if(pre != NULL)//被删除节点不是头节点
{
pre->next = p->next;
}
delete p;
p = NULL;
}
}