LeetCode（138） Copy List with Random Pointer

题目

A linked list is given such that each node contains an additional random pointer which could point to any node in the list or null.

Return a deep copy of the list.

分析

实现一个链表的深拷贝，返回拷贝后的新链表。

若是普通链表，逐个拷贝原始链表节点并连接即可，只需O(n)的时间复杂度；但是此题特殊的是，每个节点都有一个random域可以指向该链表中的任何一个节点，所以直接复制无法处理random域，因为其指向的节点很有可能还没有创建出来。

有两种方法处理：

方法一：暴力解决，首先不处理random域，将原始链表复制一份，然后遍历每个原始链表节点，查找其random域，将新链表的对应节点链接，该方法需要O(n^2)的时间复杂度，给出的结果是TLE。

方法二：充分利用原始链表的信息，不用保存原始链表的映射关系，构建新节点时，指针做如下变化，即把新节点插入到相应的旧节点后面（参考）：



同理分两步

1、构建新节点random指针：

new1->random = old1->random->next, new2-random = NULL,
new3-random = NULL, new4->random = old4->random->next

2、恢复原始链表以及构建新链表：

old1->next = old1->next->next,  new1->next = new1->next->next

该算法时间复杂度O(N)，空间复杂度O(1)

AC代码

class Solution {
public:
//方法一：直接复制，再修改random指针
RandomListNode *copyRandomList1(RandomListNode *head) {
if (!head)
return NULL;

RandomListNode *ret = new RandomListNode(head->label), *q = ret;
RandomListNode *p = head->next;
while (p)
{
RandomListNode *tmp = new RandomListNode(p->label);
q->next = tmp;
q = q->next;

p = p->next;
}//while
q->next = NULL;

//处理原始链表的random指针
p = head, q = ret;
RandomListNode *idx1 = head, *idx2 = ret;
while (p)
{
if (p->random == NULL)
q->random = NULL;
else{
idx1 = head;
idx2 = ret;
while (p->random->label != idx1->label)
{
idx1 = idx1->next;
idx2 = idx2->next;
}//while
q->random = idx2;
}//else
p = p->next;
q = q->next;
}//while
return ret;
}

//方法二：充分利用原始链表信息，在每个节点后复制添加
RandomListNode *copyRandomList(RandomListNode *head) {
if (!head)
return NULL;

//首先，复制原始的节点，连接自身后面
RandomListNode *p = head;
while (p)
{
RandomListNode *tmp = new RandomListNode(p->label);
//保存后续节点
RandomListNode *r = p->next;

tmp->next = r;
p->next = tmp;

p = r;
}//while

//然后，将添加的节点random 链接到原始节点random的下一个位置
p = head;
while (p)
{
RandomListNode *q = p->next;
if (p->random == NULL)
q->random = NULL;
else{
q->random = p->random->next;
}//else
//处理下一个原始节点
p = q->next;
}//while

//最后，恢复原始链表，得到新链表
RandomListNode *ret = head->next;

p = head;
RandomListNode *q = head->next;
while (q->next)
{
p->next = q->next;
p = q;
if (q->next)
q = q->next;
}
p->next = NULL;
q->next = NULL;
return ret;
}

};

GitHub测试程序源码