题目:
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
解题思路
常规思路:首先想到的可能是先复制复杂指针的 label 和 next,然后再查找 random并更新。查找 random又分为两种:一种是每次都从头查找,时间复杂度为O(n^2);另一种是空间换时间,复制 label和 next的同时建立一个 hash表来存放新旧复杂指针的对应关系,所以后续只需一步就能找到 random,算法时间复杂度为O(n)。
我们这里将复杂链表的复制过程分解为三个步骤,在写代码的时候我们每一步定义一个函数,这样每个函数完成一个功能,整个过程的逻辑也就非常清晰明了了:
- 第一步:复制复杂指针的 label和 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){
//链表为空
if(!pHead) return nullptr;
//第一步:复制节点,插入到原节点后面
RandomListNode* pNode = pHead;
while(pNode){
//开辟内存空间
RandomListNode* pClone = new RandomListNode(0);
//节点完全复制
pClone->label = pNode->label;
pClone->next = pNode->next;
//插入原节点后面
pNode->next = pClone;
//进入下一个节点复制插入
pNode = pClone->next;
}
//第二步:设置复制节点的random指针
//重头再来
pNode = pHead;
while(pNode){
//指向复制的新节点
RandomListNode* pClone = pNode->next;
//如果原节点random指针不为空
if(pNode->random) pClone->random = pNode->random->next;
//进入下一个节点的设置
pNode = pClone->next;
}
//第三步:拆分链表,将链表拆分为原链表和复制后的链表
pNode = pHead;
RandomListNode* pCloneHead = pNode->next;
RandomListNode* pClone = pNode->next;
while(pNode){
pNode->next = pClone->next; //奇数位连接
pNode = pNode->next; //移到下一个pNode
if(pClone->next){
pClone->next = pNode->next; //偶数位连接
pClone = pClone->next;//移到下一个pClone
}
}
//返回复制后复杂链表的head
return pCloneHead;
}
};