剑指Offer算法实现之二十六：复杂链表的复制

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

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

思路：

原链表的两层关系：后继与兄弟。
遍历一次并构建新链表是不可能的：构建过程中，原节点的兄弟指针指向的节点对应的新节点可能尚未构建。
若构建新的链表与源链表”分离“，则需要”查找“来构建兄弟关系。若m_nValue可重复，则更增加了复杂性，且算法时间效率不高。
使用方法（分层解决依赖、保存对应关系）：
① 遍历源链表，同时创建新节点并将m_nValue拷贝过来。将新节点挂在源节点的m_pSibling下，新节点的m_pNext指向源节点的兄弟
② 再次遍历源链表，构建新链表的兄弟关系
③ 再次遍历源链表，将两条链表剥离：即恢复源链表的兄弟关系。

编译环境：ArchLinux+Clang3,3，C++11

实现一：

#include <iostream>
using namespace std;

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

ComplexListNode *clone(ComplexListNode *pHead)
{
    if (!pHead) return nullptr;
    /** 复制节点，新节点挂在原节点的pSibling下，并且原节点的Sibling关系用新节点的Next保存 **/
    ComplexListNode *pNode = pHead;
    while (pNode) {
        pNode->m_pSibling = new ComplexListNode {
                                        pNode->m_nValue,
                                        pNode->m_pSibling,
                                        nullptr 
                                    };
        pNode = pNode->m_pNext;
    }
    /** 复制新节点的Sibling关系 **/
    pNode = pHead;
    while (pNode) {
        pNode->m_pSibling->m_pSibling = pNode->m_pSibling->m_pNext 
                        ? pNode->m_pSibling->m_pNext->m_pSibling
                        : nullptr;
        pNode = pNode->m_pNext;
    }
    /** 将新节点链接 **/
    pNode = pHead;
    ComplexListNode *pClonedHead = nullptr;
    ComplexListNode *pClonedPrev = nullptr;
    while (pNode) {
        if (pClonedHead) {
            pClonedPrev->m_pNext = pNode->m_pSibling;
        } else { // 头结点
            pClonedHead = pNode->m_pSibling;
        }
        pClonedPrev = pNode->m_pSibling;
        pNode->m_pSibling = pNode->m_pSibling->m_pNext; // 恢复原链表的sibling关系
        pNode = pNode->m_pNext;
    }
    return pClonedHead;
}

void printList(ComplexListNode *pHead)
{
    while (pHead) {
        cout << pHead->m_nValue;
        if (pHead->m_pSibling)
            cout << '(' << pHead->m_pSibling->m_nValue << ')';
        cout <<' ';
        pHead = pHead->m_pNext;
    }
    cout << endl;
}
int main()
{
    ComplexListNode p[5] {{1},{2},{3},{4},{5}};
    for (int i = 0; i <= 3; i++) {
        p[i].m_pNext = &p[i+1];
    }
    p[0].m_pSibling = &p[2];
    p[1].m_pSibling = &p[4];
    p[3].m_pSibling = &p[1];
    printList(p);

    ComplexListNode *pClonedHead = clone(&p[0]);
    printList(pClonedHead);
}