题目:
给定单向链表的头指针和一个结点指针,定义一个函数在 O(1)时间删除该结点。链表结点与函数的定义如下:
代码:
void DeleteNode(ListNode** pListHead,ListNode* pToBeDeleted) {
if (!pListHead || !pToBeDeleted)return;
ListNode* p = pToBeDeleted;
if (p->next) {
p->val = p->next->val;
p->next = p->next->next;
}
else {
ListNode* head = *pListHead;
if (!head->next) {
*pListHead = NULL;
}
else {
while (head->next && head->next->next) {
head = head->next;
}
head->next = NULL;
}
}
}
int main() {
ListNode* l1 = new ListNode(1);
ListNode* l2 = new ListNode(2);
ListNode* l3 = new ListNode(3);
ListNode* l4 = new ListNode(4);
ListNode* l5 = new ListNode(5);
l2->next = l3;
l3->next = l4;
l4->next = l5;
ListNode** pListHead = &l1;
ListNode* pToBeDeleted = l1;
DeleteNode(pListHead, pToBeDeleted);
cout << pListHead;
return 0;
}
思路:
如果删除的节点不是尾结点,那么将它下一个节点的值复制到它身上,然后删除它的下一个节点;
如果删除的节点是尾结点,那么从头结点开始遍历,遍历到它的前一个节点,然后删除尾结点;
如果删除的节点是尾结点且它又是头结点,那么直接将头结点置空。