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