题目:请实现函数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);
}