题目:
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
思路:
主要分为三步:
1. 先复制链表中的每个节点,并接到相应节点之后
2. 给原来有random指向的节点的复制节点连接random项
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)
{
CloneItem(pHead);
ConecRandom(pHead);
return SplitClone(pHead);
}
/*-----------克隆链表的节点-----------*/
void CloneItem(RandomListNode* pHead)
{
RandomListNode* pNode = pHead;
while(pNode != nullptr)
{
RandomListNode* pClone = new RandomListNode(pNode->label);
pClone->next = pNode->next;
pClone->random = nullptr;
pNode->next = pClone;
pNode = pClone->next;
}
}
/*-----------给克隆的节点连接random项-----------*/
void ConecRandom(RandomListNode* pHead)
{
RandomListNode* pNode = pHead;
while(pNode != nullptr)
{
RandomListNode* pClone = pNode->next;
if(pNode->random != nullptr)
pClone->random = pNode->random->next;
pNode = pClone->next;
}
}
/*-----------将克隆链表和原来的链表分离开-----------*/
RandomListNode* SplitClone(RandomListNode* pHead)
{
RandomListNode* pNode = pHead;
RandomListNode* pCloneHead = nullptr;
RandomListNode* pClone = nullptr;
if(pNode != nullptr)
{
pCloneHead = pClone = pNode->next;
pNode->next = pClone->next; // 因为pNode不为空,pClone是复制的pNode,所以一定不为空,可以访问next
pNode = pNode->next;
}
while(pNode != nullptr)
{
pClone->next = pNode->next;
pClone = pClone->next;
pNode->next = pClone->next;
pNode = pNode->next;
}
return pCloneHead;
}
};