劍指offer題解(複雜連結串列的複製)
題目描述
輸入一個複雜連結串列(每個節點中有節點值,以及兩個指標,一個指向下一個節點,另一個特殊指標指向任意一個節點),返回結果為複製後複雜連結串列的head。(注意,輸出結果中請不要返回引數中的節點引用,否則判題程式會直接返回空)
思路
新建一個連結串列,先不管random指標,根據next指標將原連結串列複製到新的連結串列中,然後遍歷新連結串列和原連結串列,依次查詢原連結串列中每個節點的random指標,新連結串列與原連結串列執行相同操作,最後確定當前節點的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)
{
RandomListNode *chead=NULL;
if(pHead==NULL)
return chead;
chead=new RandomListNode(0);
RandomListNode *p=pHead,*q=chead;
while(p->next!=NULL)
{
q->label=p->label;
p=p->next;
RandomListNode *t=new RandomListNode(0);
q->next=t;
q=t;
}
q->label=p->label;
p=pHead,q=chead;
RandomListNode *t1,*t2;
while(q!=NULL)
{
t1=pHead,t2=chead;
while(t1!=p->random)
{
t1=t1->next;
t2=t2->next;
}
q->random=t2;
p=p->next;
q=q->next;
}
return chead;
}
};