题目:给定单向链表的头指针和一个节点指针,定义一个函数在O(1)时间删除该节点。链表节点与函数定义如下:
//单向链表的节点定义
struct ListNode{
int m_value; //节点存储的值
ListNode *m_next; //节点指针
void show(); //打印所有节点值
};
解题思路:
- 如果从头至尾找到节点并删除需要O(n)时间。
- 我们直接把要删除节点的下一个节点内容复制到待删除的节点,然后把待删除的节点指针指向它的下个节点,此时删除下一个节点, 效果和删除当前结点是一样的。
- 采用此种方法时,如果待删除节点是尾节点,仍需由头至尾遍历删除。
- 如果只有一个节点,要记得把头节点设置为NULL。
测试用例:
int main(){
//创建一个链表,只有一个节点
ListNode *L1 = new ListNode;
L1 = NULL;
addToTail(&L1, 111);
//显示L1
std::cout << "L1: ";
L1->show(); //Output: 111
//删除L1唯一的节点
DeleteNode(&L1, L1);
L1->show(); //Output: list is NULL (自写的调试信息)
std::cout << std::endl;
//创建一个多结点链表
ListNode *L2 = new ListNode;
L2 = NULL;
addToTail(&L2, 1);
addToTail(&L2, 2); //末尾加入2
addToTail(&L2, 3); //末尾加入3
addToTail(&L2, 4); //末尾加入4
addToTail(&L2, 5); //末尾加入5
addToTail(&L2, 6); //末尾加入6
//显示L2
std::cout << "L2: ";
L2->show(); //Output:1,2,3,4,5,6
std::cout << std::endl;
//删除L2的头节点
DeleteNode(&L2, L2);
//显示L2
std::cout << "L2: ";
L2->show(); //Output:2, 3, 4, 5, 6
std::cout << std::endl;
//删除L2的随机一个节点
DeleteNode(&L2, L2->m_next->m_next->m_next);
std::cout << "L2: ";
L2->show(); //Output: 2, 3, 4, 6
return 0;
}
函数实现:
//本题函数实现
void DeleteNode(ListNode **listHead, ListNode *pToBeDeleted){
//判断输入
if(listHead == NULL || pToBeDeleted == NULL)
throw new std::exception("Invalid input");
if(pToBeDeleted->m_next != NULL){//如果待删除节点的下一节点不为空
//待删除节点的下一节点
ListNode *pNext = pToBeDeleted->m_next;
//复制下一节点的值到待删除节点
pToBeDeleted->m_value = pNext->m_value;
//并且将指针指向下一节点的下一节点
pToBeDeleted->m_next = pNext->m_next;
//删除下一节点,和删除待删节点效果一样. 因为原节点已被覆盖,指针也变换了位置
delete pNext;
pNext = NULL;
}else if(*listHead == pToBeDeleted){ //如果只有一个节点,删除头节点也是尾节点
delete pToBeDeleted;
pToBeDeleted = NULL;
*listHead = NULL;
}else{//否则被删除的节点是尾节点,需要由头至尾遍历删除
//副本
ListNode *pNode = *listHead;
while(pNode->m_next != pToBeDeleted){
pNode = pNode->m_next;
}
//至此已指向待删除节点(尾节点)的前一个节点,将pNode的下一节点删除,就是删除pToBeDeleted
pNode->m_next = NULL;//指定尾部
delete pToBeDeleted;
pToBeDeleted = NULL;
}
}
其他辅助方法定义:
void ListNode::show(){
ListNode *temp = this;
if(temp == NULL)
std::cout << "\nlist is NULL\n";
while(temp != NULL){
std::cout << temp->m_value << " ";
temp = temp->m_next;
}
}
//在一个链表的末尾添加一个节点
void addToTail(ListNode **head, int value){//由于会改动链表指针,所以必须把head参数设为指向指针的指针。否则出了这个函数作用域,改动便无效。
//1.首先创建要插入的节点
ListNode *pNew = new ListNode;
pNew->m_value = value; //指定值
pNew->m_next = NULL; //!!!切记指定下一节点为NULL
//2.如果传入的节点为空,可直接创键头节点
if(*head == NULL)
*head = pNew;
else{//否则3.
//定义一个临时节点
ListNode *temp = *head;
while(temp->m_next != NULL) //循环指向末尾节点, !!!切记必须指向下一节点
temp = temp->m_next;
//对末尾的下一个节点赋值,取代之(末尾)
temp->m_next = pNew;
}
}