题目描述
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
思路:
拷贝一个带有random指针的链表。
首先想到的便是,复制原链表的的每一个节点,并用next连接起来,在遍历原链表的时候,将原节点和新复制的节点的映射存进map中<Node,Node’>。然后再同时遍历原链表和新链表,此时新链表的random指针指向的元素便是原链表节点指向的节点在map中的映射,也就是A(random)->B,那么A’(random)->(map[A(random)]),因为在map中存储了原节点和新节点的对应关系,可以在O(1)的时间查找到。
还有一种方法是先遍历原链表,将复制的节点插在原链表节点的后面,也就是A->A’->B->B’->NULL,这样的话,复制的节点和原节点也就有了对应的关系,新复制的节点random指向的元素便是原节点random指向的元素的下一个元素,所有新节点random处理完毕后将这个链表再拆分成两个链表即可。
第一步:复制节点,将新创建的节点N’接到N的后面。
第二步:复制random指针
第三步:拆分链表
C++
/*
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 == nullptr) return nullptr;
RandomListNode *p = pHead;
//在每个节点A后复制一个相同的节点A',A->B->C->NULL, A->A'->B->B'->C->C'->NULL
while (p != nullptr)
{
RandomListNode *temp = new RandomListNode(p -> label);
temp -> next = p -> next;
p -> next = temp;
p = temp -> next;
}
//复制random指针,A'random指向的一定是Arandom指向的下一个元素。
p = pHead;
while (p != nullptr)
{
if (p -> random != nullptr)
p -> next -> random = p -> random -> next;
p = p -> next -> next;
}
//将链表拆分,A->A'->B->B'->C->C'->NULL,A->B->C->NULL and A'->B'->C'->NULL
p = pHead;
RandomListNode *myHead = p -> next;
RandomListNode *q = myHead;
while (p != nullptr)
{
p -> next = q -> next;
if (q -> next != nullptr)
q -> next = p -> next -> next;
p = p -> next;
q = q -> next;
}
return myHead;
}
};
参考:
https://www.cnblogs.com/silentteller/p/11931355.html