题目描述
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
剑指offer上面给的方法有点出乎意料,一般情况下想不到,这里给出正常思路的方法。
基本思想:先不考虑特殊指针,对链表进行复制,等链表复制成功后再确定每个节点特殊指针的指向。
代码如下:
/*
struct RandomListNode {
int label;
struct RandomListNode *next, *random;
RandomListNode(int x) :
label(x), next(NULL), random(NULL) {
}
};
*/
class Solution {
public:
RandomListNode* Clone(RandomListNode* pHead)
{ //要创建 pNode, pNext, pFirst, pCur, p 等五个指针
if(pHead == NULL) return NULL; //如果链表为空,返回 NULL
RandomListNode *pNode = pHead; //定义指针指向原链表头结点
RandomListNode *pFirst = new RandomListNode(0);//定义指针指向复制后链表头结点并实例化
RandomListNode *pCur = new RandomListNode(0);//定义指针指向复制后链表第二结点并实例化
RandomListNode *pNext = pNode->next;//定义指针指向原链表第二个结点
if(pNext != NULL)//如果原链表第二个结点不为空
pCur->label = pNext->label;//将原链表第二个结点的值赋给复制后链表第二结点
else
{//如果为空,说明原链表只有一个结点
pFirst->label = pNode->label;
return pFirst;//将原链表头结点的值赋给复制后链表头结点并返回
}
pFirst->label = pNode->label;//将原链表头结点的值赋给复制后链表头结点
pFirst->next = pCur;//复制后链表头结点指向下一个结点
while(pNext != NULL)//原链表的当前结点不为空
{
RandomListNode *p = new RandomListNode(0);//创建一个结点
if(pNext->next != NULL)//若原链表下一结点不为空
p->label = pNext->next->label;//p 的值等于一下结点的值
else break;//如为空则结束
pCur->next = p;//复制后链表的当前结点指向下一结点
pCur = p;//更新复制后链表的当前结点
if(pNext->next != NULL)
pNext = pNext->next;//更新原链表的当前结点
}
CopyRandom(pFirst, pHead);//确定特殊指针的指向
return pFirst;//返回复制后链表的头结点
}
void CopyRandom(RandomListNode* pFirst, RandomListNode* pHead)
{
RandomListNode* pNode = pHead;//指向原链表头结点
RandomListNode* pCur = pFirst;//指向复制后链表头结点
while(pNode != NULL && pCur != NULL)//若结点不为空
{
if(pNode->random == NULL)//若原结点的特殊指针为空
{//不需要对特殊指针处理,更新原结点与复制结点,并结束本轮循环
pNode = pNode->next;
pCur = pCur->next;
continue;
}
else
{//寻找特殊指针指向的结点需要从头结点进行遍历,所以需要再次定义指针
RandomListNode* pNode_s = pHead;
RandomListNode* pCur_s = pFirst;
while(pNode->random != pNode_s)
{//没有找到特殊指针指向的结点一直向下找,直到找到为止,方法是分别沿原链表和复制链表同时向下走
pNode_s = pNode_s->next;
pCur_s = pCur_s->next;
}
pCur->random = pCur_s;//确定特殊指针的指向
pNode = pNode->next;//更新原结点
pCur = pCur->next;//更新复制结点
}
}
}
};