您的位置:首页 > Web前端

剑指offer:复杂链表的复制

2015-11-20 16:15 465 查看
题目描述:输入一个复杂链表,每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点(可能为NULL),求输出复杂链表的拷贝(返回拷贝链表的头结点即可)
思路:一般复制一个简单链表就这么遍历一遍就好了,这个复杂链表,比简单链表多的地方就在于多了一个random的指针,也就是说在建立完简单链表之后,如何在新的链表中找到random对应的地址。我们已知的是旧的节点的地址,所以只需要用一个map,保存每一个节点旧的节点对应的新的节点的地址即可。
第一次遍历,建立简单节点,第二次遍历,对于旧链表中的每一个节点的random指针地址,从map中找到新链表中对应节点的地址,连接上就好了。
#include<iostream>
#include<map>

using namespace std;

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 NULL;
RandomListNode * res = new RandomListNode(pHead->label);
RandomListNode * tmp = res;
RandomListNode * p = pHead;
map<RandomListNode *, RandomListNode *> m;
m[p] = res;
//tmp = tmp->next;
p = p->next;
while (p)
{
//tmp = new RandomListNode(p->label);
//m[p] = tmp;
//tmp = tmp->next;
//p = p->next;
RandomListNode * newNode = new RandomListNode(p->label);
tmp->next = newNode;
tmp = newNode;
m[p] = tmp;
p = p->next;
}
p = pHead;
tmp = res;
while (p)
{
if (p->random)
tmp->random = m.find(p->random)->second;
tmp = tmp->next;
p = p->next;
}
return res;
}
};

void deleteList(RandomListNode * pHead)
{
while (pHead != NULL)
{
RandomListNode * p = pHead->next;
delete pHead;
pHead = p;
}
}

int main()
{
RandomListNode * p1 = new RandomListNode(1);
RandomListNode * p2 = new RandomListNode(2);
RandomListNode * p3 = new RandomListNode(3);
RandomListNode * p4 = new RandomListNode(4);
RandomListNode * p5 = new RandomListNode(5);
p1->next = p2; p2->next = p3;
p3->next = p4; p4->next = p5;
p5->next = NULL;
p1->random = p3; p2->random = p4;
p3->random = NULL; p4->random = p2;
p5->random = NULL;

Solution s;
RandomListNode * newHead = s.Clone(p1);

deleteList(newHead);
deleteList(p1);

return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: