输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针random指向一个随机节点),请对此链表进行深拷贝,并返回拷贝后的头结点。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
方法一:
/*
struct RandomListNode {
int label;
struct RandomListNode *next, *random;
RandomListNode(int x) :
label(x), next(NULL), random(NULL) {
}
};
*/
class Solution {
public:
RandomListNode* Clone(RandomListNode* pHead)
{
if(pHead==NULL)
{
return NULL;
}
//1.在每个节点后添加一个复制的新节点
RandomListNode *currentNode=pHead;
while(currentNode!=NULL)
{
RandomListNode *cloneNode=new RandomListNode(currentNode->label);
RandomListNode *nextNode=currentNode->next;
currentNode->next=cloneNode;
cloneNode->next=nextNode;
currentNode=nextNode;
}
//2复制randam指针
currentNode=pHead;
while(currentNode!=NULL)
{
currentNode->next->random=currentNode->random==NULL?NULL:currentNode->random->next;
currentNode=currentNode->next->next;
}
//3.拆分
currentNode=pHead;
RandomListNode *pcloneNode =pHead->next;
while(currentNode!=NULL)
{
RandomListNode *cloneNode =currentNode->next;
currentNode->next=cloneNode->next;
cloneNode->next=currentNode->next==NULL?NULL:currentNode->next->next;
currentNode=currentNode->next;
}
return pcloneNode;
}
};
方法二:
//复杂链表的复制
RandomListNode *copyRandomList(RandomListNode *head)
{
std::map<RandomListNode *, int>node_map;
std::vector<RandomListNode*> node_vec;
RandomListNode *ptr = head;
int i = 0;
while (ptr)
{
node_vec.push_back(new RandomListNode(ptr->lable));
node_map.insert(make_pair(ptr, i));
ptr = ptr->next;
i++;
}
node_vec.push_back(0);
ptr = head;
i = 0;
while (ptr)
{
node_vec[i]->next = node_vec[i + 1];
if (ptr->random)
{
int id = node_map[ptr->random];
node_vec[i]->random = node_vec[id];
}
ptr = ptr->next;
i++;
}
return node_vec[0];
}