华电北风吹
天津大学认知计算与应用重点实验室
日期:2015/10/3
题目描述
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点)。
解析:这个的问题是复制的时候地址全部都要改变,所以特殊指针的指向不好确定,目前我采用的是对原序列上指针从根节点地址进行遍历比较,同时对复制链表进行相同的操作,知道特殊指针指向正确。
/*
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 pHead;
RandomListNode* root = new RandomListNode(pHead->label);
RandomListNode* p1 = pHead, *p2 = root;
while (p1->next != NULL)
{
p1 = p1->next;
p2->next = new RandomListNode(p1->label);
p2 = p2->next;
}
p1 = pHead;
p2 = root;
while (p1 != NULL)
{
if (p1->random != NULL)
{
RandomListNode* pt1 = pHead, *pt2 = root;
while (pt1 != p1->random)
{
pt1 = pt1->next;
pt2 = pt2->next;
}
p2->random = pt2;
}
p1 = p1->next;
p2 = p2->next;
}
return root;
}
};