复杂链表的复制

1.题目描述：
请实现函数ComplexListNode* Clone(ComplexListNode* pHead)，复制一个复杂链表。在复杂链表中，每个结点除了有一个m_pNext指针指向下一个结点外，还有一个m_pSibling指向链表中的任意结点或者NULL。结点的C++定义如下：
struct ComplexListNode
{
   int              m_nValue;
   ComplexListNode* m_pNext;
   ComplexListNode* m_pSibling;
}
2.分析：
1）方法一：第一步是复制原始链表上的每一个结点，并用m_pNext链接起来；第二步是设置每个结点的m_pSibling指针。时间复杂度为O(n^2)。
2）方法二：利用一个哈希表，把<N,N’>的配对信息放到一个哈希表中，这样我们可以在O(1)时间内根据S找到S’。相当于用空间换时间。时间复杂度降为O(n)。
3）方法三：第一步，把N’链接在N的后面；第二步，设置复制出来的结点的m_pSibling；第三步：将这个长链表拆分为两个链表。此方法实现了在不用辅助空间的情况下实现O(n)的时间复杂度。
3.源代码：

#include <iostream>
#include <assert.h>
using namespace std;

struct ComplexListNode
{
    int               m_nValue;
    ComplexListNode*  m_pNext;
    ComplexListNode*  m_pSibling;
};

//第一步：根据原始链表的每个结点N创建对应的N’。
void CloneNodes(ComplexListNode* pHead)
{
    assert(pHead);
    ComplexListNode* pNode = pHead;
    while(pNode!=NULL)
    {
        ComplexListNode* pCloned = new ComplexListNode();
        pCloned->m_nValue = pNode->m_nValue;
        pCloned->m_pNext = pNode->m_pNext;
        pCloned->m_pSibling = NULL;
        
        pNode->m_pNext = pCloned;
        
        pNode = pCloned->m_pNext;
    }
}

//第二步：设置复制出来的结点的m_pSibling。
void ConnectSiblingNodes(ComplexListNode* pHead)
{
    assert(pHead);
    ComplexListNode* pNode = pHead;
    while (pNode != NULL)
    {
        ComplexListNode* pCloned = pNode->m_pNext;
        if(pNode->m_pSibling != NULL)
        {
            pCloned->m_pSibling = pNode->m_pSibling->m_pNext;
        }
        pNode = pCloned->m_pNext;
    }
}

//第三步：把这个长链表拆分成两个链表。
ComplexListNode* ReconnectNodes(ComplexListNode* pHead)
{
    assert(pHead);
    ComplexListNode* pNode = pHead;
    ComplexListNode* pClonedHead = NULL;
    ComplexListNode* pClonedNode = NULL;
    
    if(pNode!=NULL)
    {
        pClonedHead = pClonedNode = pNode->m_pNext;
        pNode->m_pNext = pClonedNode->m_pNext;
        pNode = pNode->m_pNext;
    }
    while(pNode!=NULL)
    {
        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);
}

ComplexListNode* CreateLinkList()
{
    int length = 0;
    int sblingData = 0;
    cout<<"请输入链表的长度:";
    cin >> length;
    //创建结点
    cout<<"请输入链表中每个结点的数据(以空格分隔开):";
    ComplexListNode *pHead = NULL;
    ComplexListNode *pNew = NULL;
    ComplexListNode *pLast = NULL;
    for(int i = 0;i<length;i++)
    {
        pNew = new ComplexListNode();
        cin>>pNew->m_nValue;
        pNew->m_pNext = NULL;
        pNew->m_pSibling = NULL;
        
        if(pHead == NULL)
        {
            pHead = pNew;
        }
        else
        {
            pLast->m_pNext = pNew;
        }
        pLast = pNew;
    }
    //创建兄弟
    ComplexListNode* pCur = pHead;
    ComplexListNode* pCurTmp = pHead;
    cout<<"请输入链表中每个结点的兄弟结点的值(以空格分隔开):";
    while(pCur)
    {
        cin>>sblingData;//接受兄弟的下标
        pCurTmp = pHead;
        if(sblingData == 0)
        {
            pCur->m_pSibling = NULL;
        }
        else
        {
            while(pCurTmp && sblingData != pCurTmp->m_nValue)
            {
                pCurTmp = pCurTmp->m_pNext;
            }
            assert(pCurTmp);
            pCur->m_pSibling = pCurTmp;
        }
        pCur = pCur->m_pNext;
    }
    return pHead;
}

//从头到尾打印链表
void printLink(ComplexListNode *pHead)
{
    ComplexListNode *pCur = pHead;
    while (pCur) {
        cout<<pCur->m_nValue<<" ";
        if(pCur->m_pSibling)
        {
            cout<<pCur->m_pSibling->m_nValue<<endl;
        }
        else
        {
            cout<<0<<endl;
        }
        pCur=pCur->m_pNext;
    }
    cout<<endl;
}

int main(int argc,char *argv[]){

    ComplexListNode* pHead = NULL;
    ComplexListNode* pNewHead = NULL;
    pHead = CreateLinkList();
    cout<<"原始链表为:"<<endl;
    printLink(pHead);
    pNewHead = Clone(pHead);
    cout<<"复制链表为:"<<endl;
    printLink(pNewHead);
    return 0;
}

4.运行效果：

请输入链表的长度:5
请输入链表中每个结点的数据(以空格分隔开):1 2 3 4 5
请输入链表中每个结点的兄弟结点的值(以空格分隔开):3 5 3 2 5
原始链表为:
1 3
2 5
3 3
4 2
5 5

复制链表为:
1 3
2 5
3 3
4 2
5 5

Program ended with exit code: 0