struct ListNode
{
int val;
ListNode* next;
};
void DeleteNode(ListNode** pListHead, ListNode* pToBeDeleted);
一般情况下,我们需要从头开始查找(时间复杂度为 O(n) ),找到被删除节点的前一个结点。正如题目所言,是否一定需要得到被删除节点的前一个结点呢?答案是否定。我们可以找到一种等价做法,如果我们把下一个节点的内容(无非是 val 和指向下一个元素的指针)覆盖自身的内容,再把下一个结点删除。
这里需要注意两个问题,如果被删除的节点位于链表的尾部,那么他就没有下一个结点(next == NULL),此时值得从节点开始遍历,找到倒数第二个节点。
如果链表只有一个节点,而我们又要删除链表的头结点(也是尾节点),需要将链表的头结点置为 NULL。
void DeleteNode(ListNode** pListHead, ListNode* pToBeDeleted)
{
if (!pListHead || !pToBeDeleted)
return;
if (pToBeDeleted->next != NULL)
{
ListNode* pNext = pToBeDeleted->next;
pToBeDeleted->val = pNext->val;
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;
}
}