题目:
给定链表的头指针和一个结点指针,定义一个函数在O(1)时间删除该结点。
思路:
1.常规做法:找到待删除结点的前一个结点。时间复杂度为O(N)不符合要求。
2结点覆盖:。所以我们可以把下一个结点的内容复制到要删除的结点,然后删除下一个结点,效果等同,且时间复杂度为O(1)。
**(注意:如果待删除结点为尾结点,我们需要用常规做法来解决,不能忽视该特殊情况,但是时间复杂度一平均依旧是O(1),可以放心。)
代码如下,有不足欢迎指正啦。
#include<iostream>
using namespace std;
//结构。
struct ListNode
{
int val;
ListNode* next;
ListNode(int x):val(x),next(NULL){}
};
class solution {
public:
ListNode* deleteSpecialnode(ListNode* head, ListNode* specialNode)
{
//特殊情况
if (!head||!specialNode) return head;
//待删除节点为尾结点
if (!specialNode->next)
{
//待操作链表仅有一个结点,头结点也是尾结点
if (head == specialNode)
return NULL;
//头结点非尾结点的情况
else
{
ListNode* tempHead = head;
//找到待删除结点的前一个结点。
while (tempHead->next != specialNode)
{
tempHead = tempHead->next;
}
tempHead->next = NULL;
delete specialNode;//删除尾结点
specialNode = NULL;//避免尾结点成为野指针。
return head;
}
}
//待删除结点非尾结点,令待删除结点覆盖其下一个结点。
//好处是不用找到待删除结点的前一个结点,时间复杂度O(1)
ListNode* afterspecialNode = specialNode->next;
specialNode->val = afterspecialNode->val;//结点值覆盖
specialNode->next = afterspecialNode->next;//更新指针
delete afterspecialNode;
afterspecialNode = NULL;
return head;
}
};
ListNode* creatList()
{
ListNode* prehead = new ListNode(-1);
ListNode* dummy = prehead;
cout << "输入链表,(通过“-1”终止输入):" << endl;
while (true)
{
int value;
cin >> value;
cout << " -> " ;
if (value == -1)
break;
ListNode* donecreatList = new ListNode(value);
dummy->next = donecreatList;
dummy = dummy->next;
}
cout << endl;
return prehead->next;
}
void printList(ListNode* head)
{
ListNode* forprintList;
forprintList = head;
while (forprintList)
{
cout << " "<<forprintList->val;
forprintList = forprintList->next;
}
cout << endl;
}
int main()
{
solution solute;
ListNode* head = creatList();
cout << "给定链表:" << endl;
printList(head);
head=solute.deleteSpecialnode(head, head->next->next);
cout << "返回链表:" << endl;
printList(head);
return 0;
}