题目:请实现函数ComplexListNode * Clone(ComplexListNode* pHead),复制一个复杂链表。在复杂链表中,每个节点除了有一个m_pNext指针指向下一个节点,还有一个m_pSibiling指针指向链表中的任意节点或者nullptr。节点的C++定义如下:
struct ComplexListNode
{
int m_nValue;
ComplexListNode* m_pNext;
ComplexListNode* m_pSibling;
}
第一种思路:
把复制过程分为两步:第一步是复制原始链表上的每个节点,并用m_pNext链接起来;第二步是设置每个节点的m_pSibling指针。假设原始链表中的某个节点N的m_pSibling指针指向S节点,由于S在链表中可能在N的前面也可能在N的后面,所以要定位S的位置需要从原始链表的头节点开始找。如果从原始节点的头节点开始沿着m_pNext经过s步找到节点S,那么在复制链表节点N'的m_pSibling(记为S')离复制链表的头节点的距离也是沿着m_pNext指针s步。用这种办法可以为复制链表上的每个节点设置m_pSibling指针。
对于一个含有n个节点的链表,由于定位每个节点的m_pSibling都需要从链表头节点开始经过O(n)步才能找到,因此总的时间复杂度O(n的平方)。
第二种思路:
(由于第一种时间主要花费在定位节点的m_pSibling上面,基于此进行优化)
第 一步仍然是复制原始链表上每个节点N创建N',然后把这些创建出来的节点用m_pNext链接起来。同时我们把<N,N'>的配对信息放到一个哈希表中;第二步还是设置复制链表上每个节点的m_pSibling。如果在原始链表中节点的N的m_pSibling指向节点S,那么在复制链表中,对应的N'应该指向S'。由于有了哈希表,我们可以用O(1)时间根据S找到S'。
第二种方法相当于用空间换时间。对于有n个节点的链表,我们需要一个大小为O(n)的哈希表。
第三种方法:
第一步仍然是根据原始链表的每个节点N创建对应的N',这一次N'链接到N的后面。
注意:复制原始链表的节点N并创建新节点N',再把N'链接到N的后面;
完成这一步的代码如下:
void CloneNodea(ComplexListNode* pHead)
{
ComplexListNode* pNode=pHead;
while(pNode!=nullptr)
{
ComeplexListNode* pCloned=new ComplexListNode();
pClone-m_nValue=pNode->m_nValue;
pClone-m_pNext=pNode->m_pNext;
pClone-m_pSibling=nullptr;
pClone-m_pNext=pCloned;
pNode=pCloned->m_pNext;
}
}
第二步设置复制出来的节点m_pSibling。假设原始链表上的N的m_pSibling指向节点S,那么其对应复制出来的N'是N的m_pNext指向的节点,同样S'也是S的m_pNext指向的节点。
注意:如果原始链表上的节点N的m_pSibling指向S,则它对应的复制节点N'的m_pSibling指向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;
pNode->m_pNext=pClonedNode->m_pNext;
pNode=pNode->m_pNext;
}
while(pNode!=nunllptr)
{
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)
{
CloneNode(pHead);
ConnectSiblingNodes(pHead);
return ReconnectNodes(pHead);
}
完整代码:https://github.com/zhedahht/CodingInterviewChinese2/tree/master/35_CopyComplexList