版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/N1neDing/article/details/82534073
题目描述:
题目:给定单向链表的头指针和一个结点指针,定义一个函数在O(1)时间删除该结点。
解题思路:
如果从头到尾遍历找到目标节点位置,所需时间复杂度为O(n);
直接将目标节点的后一位节点的值赋值给删除目标节点,然后将待删除节点指向node->next->next的位置,即相当于使其后一位替代该位,达到删除的效果。
因为需要待删除节点之后的再之后的节点,所以需要考虑待删除节点为尾节点的情况,此时需要使用遍历链表的方法;
以及当删除节点即为首节点又为尾节点时,直接将节点删除即可。
参考源码:
#include <iostream>
using namespace std;
struct ListNode
{
int m_nValue;
ListNode* m_pNext;
};
void DeleteNode(ListNode* pListHead, ListNode* pToBeDeleted)
{
if (!pListHead || !pToBeDeleted) return;
//要删除的节点不是尾节点,时间复杂度为O(1)
if (pToBeDeleted->m_pNext != NULL)
{
ListNode* pNext = pToBeDeleted->m_pNext;
pToBeDeleted->m_nValue = pNext->m_nValue;
pToBeDeleted->m_pNext = pNext->m_pNext;
delete pNext;
pNext = NULL;
}
//链表只有一个节点,删除的是头尾节点
else if (pListHead == pToBeDeleted)
{
delete pListHead;
pListHead = NULL;
pToBeDeleted = NULL;
}
//删除尾节点
else if (!pToBeDeleted->m_pNext)
{
ListNode* pNode = pListHead;
while (pNode->m_pNext != pToBeDeleted)
{
pNode = pNode->m_pNext;
}
pNode->m_pNext = NULL;
delete pToBeDeleted;
pToBeDeleted = NULL;
}
}