面试题35:复杂链表的复制

题目:请实现函数ComplexListNode* Clone(ComplexListNode* pHead),复制一个复杂链表,在复杂链表中,每个节点除了有一个m_pNext指针指向下一个节点,还有一个m_pSibling指针指向链表中的任意节点或者nullptr。节点的C++定义如下:

struct ComplexListNode
{
     
    int m_nValue;
    ComplexListNode* m_pNext;
    ComplexListNode* m_pSibling;
};
//第一步:根据原始链表的每个节点N创建对应的N‘,把N’链接在N的后面
void CloneNodes(ComplexListNode* pHead)
{
     
    ComplexListNode* pNode=pHead;
    while(pNode!=nullptr)
    {
     
        ComplexListNode* pCloned=new ComplexListNode();
        pCloned->m_nValue=pNode->m_nValue;
        pCloned->m_pNext=pNode->m_pNext;
        pCloned->m_pSibling=nullptr;

        pNode->m_pNext=pCloned;
        pNode=pCloned->m_pNext;
    }
}

//第二步:假设原始链表上的N的m_pSibling指向节点S,那么其对应复制出来的N'是N的m_pNext指向的节点,同样的S‘也是S的m_pNext指向的节点
void ConnectSiblingNodes(ComplexListNode* pHead)
{
     
    ComplexListNode* pNode=pHead;
    while(pNode!=nullptr)
    {
     
        ComplexListNode* pCloned=pNode->m_pNext;
        if(pNode->m_pSibling!=nullptr)
        {
     
            pCloned->m_pSibling=pNode->m_pSibling->m_pNext;
        }
        pNode=pCloned->m_pNext;
    }
}

//第三步:把偶数位置的节点用m_pNext链接起来就是复制出来的链表
ComplexListNode* ReconnectNodes(ComplexListNode* pHead)
{
     
    ComplexListNode* pNode=pHead;
    ComplexListNode* pClonedHead=nullptr;
    ComplexListNode* pClonedNode=nullptr;

    if(pNode!=nullptr)
    {
     
        pClonedHead=pClonedNode=pNode->m_pNext;
        pNode->m_pNext=pClonedNode->m_pNext;
        pNode=pNode->m_pNext;
    }

    while(pNode!=nullptr)
    {
     
        pClonedNode->m_pNext=pNode->m_pNext;
        pClonedNode=pClonedNode->m_pNext;
        pNode->m_pNext=pClonedNode->m_pNext;
        pNode=pNode->m_pNext;
    }

    return pClonedHead;
}

ComplexListNode* Clone(ComplexListNode* pHead)
{
     
    CloneNodes(pHead);
    ConnectSiblingNodes(pHead);
    return ReconnectNodes(pHead);
}

你可能感兴趣的:(剑指Offer,链表,指针,单链表)