C++ 复杂链表的复制

struct ComplexNode
{
    ComplexNode(const int& d)
        :_data(d)
        ,_next(NULL)
        ,random(NULL)
    {}
    int _data;            //数据域
    ComplexNode* _next;               //指向下一节点
    ComplexNode* _random;             //指向随机节点
};

void CloneNodes(ComplexNode* pHead)		
{
	ComplexNode* cur = pHead;
	while(cur)
	{
		ComplexNode* NewHead = new ComplexNode();	//开辟新节点
		NewHead->_data = cur->_data;
		NewHead->_next = cur->_next;
		NewHead->_random = NULL;

		cur->_next = NewHead;		//连接新节点

		cur = NewHead->_next;		//跳转到下一个要复制的节点
	}
}

void UpdateNewNodeRandom(ComplexNode* pHead)
{
	ComplexNode* cur = pHead;
	while(cur)
	{
		ComplexNode* next = cur->_next;			//指向新节点

		if(cur->_random != NULL)	//优化：随机指针不为空时，复制
		{
			next->_random = cur->_random->_next;	//复制随机指针
		}

		cur = next->_next;			//下一个要复制的节点
	}
}

ComplexNode* DisconnectNewNode(ComplexNode* pHead)
{
	ComplexNode* cur = pHead;		//指向原来的节点
	ComplexNode* next = NULL;		//指向复制出来的节点
	ComplexNode* pNewHead = NULL;	//新链表的头节点

	if(cur != NULL)
	{
		pNewHead = next = cur->_next;	//指向新链表的头
		cur->_next = next->_next;		//将新节点分离
		cur = next->_next;				
	}

	while(cur)
	{
		next->_next = cur->_next;		//往复制出的链表上面连接新节点

		next = next->_next;				//分离新节点
		cur->_next = next->_next;

		cur = next->_next;				
	}

	return pNewHead;					//返回新链表的头结点

}

ComplexNode<T>* CopyComplexLinkList(ComplexNode<T>* pHead)
{
	if(pHead != NULL)					//判空
	{
		CloneNodes<T>(pHead);			//复制节点并连接在其后面	
		UpdateNewNodeRandom(pHead);		//更新新节点的随机指针
		return DisconnectNewNode<T>(pHead);/    /拆分链表并返回新链表的头结点
	}
	return NULL;
}

ComplexNode* CreatList()
{
	ComplexNode* Node1 = new ComplexNode(1);    //创建节点
	ComplexNode* Node2 = new ComplexNode(2);
	ComplexNode* Node3 = new ComplexNode(3);
	ComplexNode* Node4 = new ComplexNode(4);
	ComplexNode* Node5 = new ComplexNode(5);
	
	Node1->_next = Node2;       //连接节点
	Node2->_next = Node3;
	Node3->_next = Node4;
	Node4->_next = Node5;

	Node1->_random = Node3;    //置_random
	Node2->_random = Node4;
	Node3->_random = Node1;
	Node4->_random = Node5;
	Node5->_random = Node2;

	return Node1;            //返回头
}

void Print(ComplexNode* _head)
{
	ComplexNode* cur = _head;
	while(cur)
	{
		cout<<"("<<cur->_data<<","<<cur->_random->_data<<")"<<"->";
		cur = cur->_next;
	}
	cout<<"Nvl."<<endl;
}

void Test()
{
	ComplexNode* head = CreatList();
	Print(head);
	cout<<"---------------------------------------"<<endl;
	ComplexNode* NewHead = CopyList(head);
	Print(NewHead);
}