1.题目要求
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
链表形式如下:
struct RandomListNode {
int label;
struct RandomListNode *next, *random;
RandomListNode(int x) :
label(x), next(NULL), random(NULL) {
}
};
2.基本思路
思路来自<剑指offer>。设原始链表头指针为pHead。
基本思路分三步:
- 在PHead链表中,每一个节点的后面插入一个一模一样的复制节点,所谓一模一样,指的是其random指针指向一样,label一样。注意是插入在pHead链表之中,原始节点之后。
- 更改每一个复制节点的random指针的指向,其不应该指向pHead之中的原始节点,而应该指向pHead中的复制节点(复制节点构成新的复制链表)即对每一个复制节点实现如下操作,
原始节点->random=原始节点->random->next.
(因为每一个原始节点之后都有一个复制节点) - 将复制节点从pHead链表中分离出来
3.代码
class Solution {
public:
RandomListNode* Clone(RandomListNode* pHead)
{
if(pHead==NULL)
return NULL;
RandomListNode *p=pHead,*p1;
RandomListNode *head;
//每一个节点之后增加一个复制节点
while(p!=NULL){
RandomListNode* r1=new RandomListNode(p->label);
p1=p->next;
r1->next=p1;
p->next=r1;
r1->random=p->random;
p=p1;
}
//更改每一个复制节点的random指针的指向,使其指向PHead中的复制节点
p=pHead->next;
while(p!=NULL && p->next!=NULL){
if(p->random)
p->random=p->random->next;
p=p->next->next;
}
if(p->random)
p->random=p->random->next;
// split
head=pHead->next;
p=pHead;
p1=pHead->next;
while(p1->next!=NULL && p->next!=NULL){
p->next=p->next->next;
p1->next=p1->next->next;
p=p->next;
p1=p1->next;
}
p->next=NULL;
return head;
}
};