在O(1)时间删除链表结点

题目:给定单向链表的头指针和一个结点指针,定义一个函数在O(1)时间删除该结点。链表结点与函数的定义如下:

struct ListNode
{
    int value;
    ListNode *next;
};
void DeleteNode(ListNode **pListNode, ListNode *pToBeDeleted);

在单向的链表中删除一个结点,最常规的做法就是从链表的头结点开始,顺序遍历查找要删除的结点,将要删除结点的前一个结点的next指针指向要删除结点的下一个结点,并在链表中删除要删除的结点。这种思路要顺序查找,时间复杂度就是O(n)了;我们之所以要从头开始查找,是因为我们需要得到将要删除结点的前一个结点,在单向链表中没有指向前一个结点的指针,所以只好从链表的头结点开始顺序查找。

那是不是一定要找到被删除的结点的前一个结点呢?其实我们换一种思路,发现我们并不是一定需要这样。我们可以很方便的找到要删除的结点的下一个结点。如果把下一个结点的内容复制要需要删除的结点上覆盖原有的内容,再把下一个结点删除,这样相当于把当前需要删除的结点删除了。不过基于这种思路我们需要考虑一种情况,就是需要删除的结点是尾结点的时候,我们同样需要从头开始遍历得到该结点的前序结点,再完成删除操作。实现代码如下:

void DeleteNode(ListNode** pListHead, ListNode* pToBeDeleted)
{
    if(!pListHead || !pToBeDeleted)
        return;

    // 要删除的结点不是尾结点
    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 pToBeDeleted;
        pToBeDeleted = NULL;
        *pListHead = NULL;
    }
    // 链表中有多个结点,删除尾结点
    else
    {
        ListNode* pNode = *pListHead;
        while(pNode->m_pNext != pToBeDeleted)
        {
            pNode = pNode->m_pNext;            
        }
 
        pNode->m_pNext = NULL;
        delete pToBeDeleted;
        pToBeDeleted = NULL;
    }
}

对于n-1个非尾结点而言,我们可以在O(1)时间内把下一个结点的内容复制覆盖需要删除的结点,并删除下一个结点;对于尾结点而言,由于仍然需要顺序查找,时间复制度是O(n),总的时间复制度诗[(n-1)*O(1)+1*O(n)]/n,结果还是O(1),符合题目的要求。

值得注意的是,上述代码我们基于一个假设:要删除的结点确实在链表中。  由于受到O(1)的限制,我们没有花时间去判断假设是否成立,因为这样需要花费O(n)的时间。

测试用例:

功能测试:从多个结点的链表的中间删除一个结点,从有多个结点的链表中删除头结点,从有多个结点的链表中删除尾结点,从只有一个结点的链表中删除唯一的结点。

特殊输入测试:指向链表头指针的为NULL指针,指向要删除的结点为NULL指针。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值