剑指offer系列源码-复杂链表的复制

题目1524：复杂链表的复制
时间限制：1 秒内存限制：128 兆特殊判题：否提交：606解决：292
题目描述：
输入一个复杂链表（每个节点中有节点值，以及两个指针，一个指向下一个节点，另一个特殊指针指向任意一个节点）。
输入：
输入可能包含多个测试样例，输入以EOF结束。
对于每个测试案例，输入的第一行为一个整数n (1<=n<=1000)：n代表将要输入的链表元素的个数。（节点编号从1开始）。
接下来有n个数，表示链表节点中的值。
接下来有n个数Ti，Ti表示第i个节点的另一个指针指向。
Ti = 0 表示这个指针为NULL。
输出：
对应每个测试案例，
输出n行，每行有二个数，第一个代表当前节点值，第二个代表当前节点的特殊指针的值。
样例输入：
5
1 2 3 4 5
3 5 0 2 0
样例输出：
1 3
2 5
3 0
4 2
5 0

#include <iostream>
#include<stdio.h>
using namespace std;
struct ListNode{
int value;
ListNode* next;
ListNode* random_link;
};

int main(){
int n,num;
while(scanf("%d",&n)!=EOF){
ListNode* nodes = new ListNode[n+1];
for(int i=1;i<=n;i++){
scanf("%d",&nodes[i].value);
nodes[i].next = NULL;
nodes[i].random_link = NULL;
}
for(int i=1;i<=n;i++){
scanf("%d",&num);
if(num>0)
nodes[i].random_link = &nodes[num];
}
for(int i=1;i<=n;i++){
printf("%d %d\n",nodes[i].value,nodes[i].random_link!=NULL?nodes[i].random_link->value:0);
}
}
return 0;
}

#include <iostream>
#include<stdio.h>
using namespace std;
struct ListNode{
int value;
ListNode* next;
ListNode* random_link;
};
//step1 复制节点
void cloneNodes(ListNode* pHead){
ListNode* pNode = pHead;
while(pNode!=NULL){
ListNode* pClone = new ListNode();
pClone->value = pNode->value;
pClone->next = pNode->next;
pClone->random_link = NULL;

pNode->next = pClone;

pNode = pClone->next;
}
}

void connectRandomNodes(ListNode* pHead){
ListNode* pNode = pHead;
while(pNode){
ListNode* pClone = pNode->next;

if(pNode->random_link!=NULL){
pClone->random_link = pNode->random_link->next;
}
pNode = pClone->next;
}
}
ListNode* reconnentNodes(ListNode* pHead){
ListNode* pNode = pHead;
ListNode* pCloneHead = NULL;
ListNode* pCloneNode = NULL;
if(pNode!=NULL){
pCloneHead = pCloneNode = pNode->next;
pNode->next = pCloneNode->next;
pNode = pNode->next;
}
while(pNode){
pCloneNode->next = pNode-next;
pCloneNode = pCloneNode->next;
pNode->next = pCloneNode->next;
pNode = pNode->next;
}
return pCloneHead;
}
//将上面三步合起来
ListNode* clone(ListNode* pHead){
cloneNodes(pHead);
connectRandomNodes(pHead);
return reconnentNodes(pHead);
}
int main(){
return 0;
}