题目描述
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针random指向一个随机节点),请对此链表进行深拷贝,并返回拷贝后的头结点。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)。
我的思路
创建新的节点,根据原链表的next指针为新的节点逐个赋值,将新链表的next指针赋值完成后,第二次遍历复制random,复制random时,判断该复制链表哪个结点与原链表要复制结点同一位置结点的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)
{
if(pHead==NULL) return pHead;
RandomListNode* cpHead=new RandomListNode(pHead->label);
RandomListNode* p=cpHead; //p用来遍历新建成的链表
RandomListNode* q=pHead->next; //q用来遍历原链表
while(q){
RandomListNode *cnode=new RandomListNode(q->label); //这是新建立的链表结点
p->next=cnode;
p=p->next;
q=q->next;
}
p->next=NULL;
//接下来复制random指针
p=cpHead; //p为要挨个赋值random的结点
q=pHead;
RandomListNode *cnode=cpHead;
RandomListNode *node=pHead;
while(p){
cnode=cpHead;
node=pHead;
while(node){
if(node==q->random){
p->random=cnode;
break;
}
cnode=cnode->next;
node=node->next;
}
if(!node) p->random=NULL;
p=p->next;
q=q->next;
}
return cpHead;
}
};
做链表题一定一定要注意判断原链表为空的情况
参考网上的思路
具体分为三步
第一步,在原链表每一个结点A的后面复制一个值相同的结点A`, A的next为A’,A原来的next为A‘的next;
第二步,A’的random为A的random的next,将每一个被复制的重复节点的random都根据这种办法赋值;
第三步,将所有的A‘节点都从原链表中抽离出来,组成一个新的链表,返回新链表的头节点。
代码这里就不再写出了。