题目:请实现函数ComplexListNode* Clone(ComplexListNode* pHead),复制一个复杂链表。在复杂链表中,每个结点除了有一个m_pNext指针指向下一个结点外,还有一个m_pSibling 指向链表中的任意结点或者nullptr。
思路:主要分为三步,具体如下
第一步:复制原始链表的任意节点N并创建新节点N’,再把N’链接到N的后面;
第二步:如果原始链表上的节点N的m_pSibling指向S,则它对应的复制节点N’的m_pSibling指向S的复制节点S’,即S的下一个节点;
第三步:把长链表拆分成两个链表。把奇数位置的节点用m_pNext链接起来就是原始链表,把偶数位置的节点用m_pNext链接起来就是复制出来的链表。
核心代码如下:
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;
}
}
//如果原始链表上的节点N的m_pSibling指向S,则它对应的复制节点N'的m_pSibling指向S的复制节点S',即S的下一个节点
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链接起来就是原始链表,把偶数位置的节点用m_pNext链接起来就是复制出来的链表
ComplexListNode* ReconnectNodes(ComplexListNode* pHead){
ComplexListNode* pNode = pHead;
ComplexListNode* pClonedHead = nullptr;
ComplexListNode* pClonedNode = nullptr;
if(pNode != nullptr){ //头结点不为空时
pClonedHead = pClonedNode = pNode->m_pNext; //pClonedHead指向复制链表的头结点
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);
}