原始思路(最容易想到):
1.先把原始链表中每个结点都复制,并通过next指针把复制结点连接起来。(特别简单)
2.然后每个结点设置random指针,这里对于一个含有n个结点的链表,由于定位每个结点的random都需要从链表头结点开始经过O(n)时间找到,所以总的时间复杂度为O(n²)。
改进的思路:
第一步:
在原始链表的每个结点N后添加当前结点的copy,并把它们连接起来。
第二步:
设置复制结点的random指针。
第三步:
把第二步得到的链表拆分成两个链表,奇数位置上的结点组成原始链表,偶数位置上的结点组成复制出来的新链表。
这样,总的时间复杂度为O(n)。
思路理解后,很快写出代码如下:
#include<iostream>
using namespace std;
struct RandomListNode {
int value;
RandomListNode*next;
RandomListNode*random;
RandomListNode(int x = 0) :value(x), next(NULL), random(NULL) {}
};
class Solution {
public:
//第一步:复制结点(即在旧链表每个结点后面复制一个结点,并连接起来)
void CloneNodes(RandomListNode*phead) {
RandomListNode*pcurrent = phead;
while (pcurrent != NULL)
{
RandomListNode*pcloned = new RandomListNode(pcurrent->value);
pcloned->next = pcurrent->next;
pcurrent->next = pcloned;
pcurrent = pcloned->next;
}
}
//第二步:处理刚才复制结点的random指针
void ConnectRandom(RandomListNode*phead) {
RandomListNode*pcurrent = phead;
while (pcurrent != NULL)
{
if (pcurrent->random != NULL)
pcurrent->next->random = pcurrent->random->next;//这里小心!
pcurrent = pcurrent->next->next;
}
}
//第三步:把链表拆成两个,返回新链表的头结点
RandomListNode*Separate(RandomListNode*phead)//奇数、偶数分开,连接起来即可
{
RandomListNode*pcurrent = phead;
RandomListNode*pcloned_node = NULL;
RandomListNode*pcloned_head = NULL;
if (pcurrent != NULL)
{
pcloned_head = pcurrent->next;
pcloned_node = pcurrent->next;
pcurrent->next = pcloned_node->next;
pcurrent = pcurrent->next;
}
while (pcurrent != NULL)
{
//交替进行
pcloned_node->next = pcurrent->next;
pcloned_node = pcloned_node->next;
pcurrent->next = pcloned_node->next;
pcurrent = pcurrent->next;
}
return pcloned_head;
}
//综合以上三步:
RandomListNode*Clone(RandomListNode*phead)
{
CloneNodes(phead);
ConnectRandom(phead);
RandomListNode*pcloned_head = Separate(phead);
return pcloned_head;
}
};
int main(void) {
system("pause");
return 0;
}