题目描述
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针random指向一个随机节点),请对此链表进行深拷贝,并返回拷贝后的头结点。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
看完题目应该很混乱,甚至还有点看不懂。
题目解析
看上面的图片,题目中说的复杂链表表示,一个节点有1-2个指针,除了指向下一个节点的指针外,还可能存在一个random指针,这个指针指向该链表中某一个节点的位置。如图所示,A节点,除了一个next指针指向B之外,还有一个指针random指向C。
设计一个算法实现复杂链表的深拷贝。
解题方法阐述
方法一:
1、遍历链表,复制每个结点,如复制结点A得到A1,将结点A1插到结点A后面;
2、重新遍历链表,复制老结点的随机指针给新结点,如A1.random = A.random.next;
3、拆分链表,将链表拆分为原链表和复制后的链表
上述方法的图示如下图所示
1
2
3
具体代码如下:
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==nullptr)return nullptr;
//复制节点,加入到复制节点的后面
//指向原来的头
auto cur=pHead;
//只要当前不是空的就复制
while(cur != nullptr){
auto lat=cur->next;
//复制当前节点
RandomListNode *temp = new RandomListNode(cur->label);
cur->next=temp;
temp->next=lat;
cur=lat;
}
//构造random
cur=pHead;
while(cur != nullptr){
if(cur->random==nullptr){
cur->next->random=nullptr;
}else{
cur->next->random=cur->random->next;
}
cur=cur->next->next;
}
//拆分链表
cur=pHead;
//新的头节点
auto head=pHead->next;
while(cur != nullptr){
auto node = cur->next;
cur->next=node->next;
node->next=node->next==nullptr?nullptr:node->next->next;
cur=cur->next;
}
return head;
}
};
方法二:
假设链表中的值没有重复
使用unordered_map记录每个节点的位置。具体操作步骤如下:
1.从头节点开始遍历,跟着->next方向建立所有的节点出来,同时用map<label, pisition>记录每个值在链表的什么位置。
2.为每个节点链接random。沿着老链表的->next遍历,根据每个节点->random的label锁定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 == nullptr)return pHead;
//unordered_map(label,位置)
map<int, int> mp;
int count=0;
//指向原来的头
auto p2=pHead;
//新建一个链表
RandomListNode *head = new RandomListNode(pHead->label);
//指向新建链表的头
auto p1=head;
mp.insert(make_pair(p2->label, count++));
//链表雏形的建立
while(p2->next != nullptr){
auto temp = new RandomListNode(p2->next->label);
p1->next = temp;
p1 = p1->next;
p2 = p2->next;
mp.insert(make_pair(p2->label, count++));
}
//一定要重新指回到头节点
p1=head;
p2=pHead;
//random指针的链接
while(p2 != nullptr){
if(p2->random!=nullptr){
auto p=head;
auto target = p2->random->label;
auto n=mp.find(target)->second;
for(int i=n;i!=0;i--)
p=p->next;
p1->random=p;
}
p1=p1->next;
p2=p2->next;
}
return head;
}
};
本次代码没有创建测试用例,可以参照我之前博客中创建测试用例的方法创建测试用例运行查看结果,创建测试用例时注意除了next指针,还要注意random指针的建立。