这道题目不仅考察思维,更重要的是考察的链表的操作。
做这道题的时候,我们一定对链表的相关操作非常熟悉,才能没有错误的写出来。
第一步我们都知道怎么做,就是按照原链表顺序指向的指针把这个链表先创建出来,然后每个节点还有一个随机指针,他们会指向不同的节点,由于对于一个节点来说,他可能指向它之前的节点,也可能指向它之后的节点。所以如果一个一个节点找的话,复杂度非常高,会达到O(n2),所以我是看了书之后的写法:就是把新创建的链表的每一个节点都放在原链表每一个节点的后面,这样原链表的自由指针指向的节点的下一个节点,就是我们要找的新创建的链表的节点的自由节点应该指向的节点。
/*
// Definition for a Node.
class Node {
public:
int val;
Node* next;
Node* random;
Node(int _val) {
val = _val;
next = NULL;
random = NULL;
}
};
*/
class Solution {
public:
void CloneNodes(Node* pHead)
{
Node* pNode=pHead;
while(pNode!= nullptr)
{
Node* pCloned=(Node*)malloc(sizeof(Node));
pCloned->val=pNode->val;
pCloned->next=pNode->next;
pCloned->random= nullptr;
pNode->next=pCloned;
pNode=pCloned->next;
}
}
void ConnectRandom(Node* pHead)
{
Node* pNode=pHead;
while(pNode!= nullptr)
{
Node* pCloned=pNode->next;
if(pNode->random!= nullptr)
{
pCloned->random=pNode->random->next;
}
pNode=pCloned->next;
}
}
Node* ReconnectNodes(Node* pHead)
{
Node* pNode=pHead;
Node* pClonedHead= nullptr;
Node* pClonedNode= nullptr;
if(pNode)
{
pClonedHead=pClonedNode=pNode->next;
pNode->next=pClonedHead->next;
pNode=pNode->next;
}
while(pNode!= nullptr)
{
pClonedNode->next=pNode->next;
pClonedNode=pClonedNode->next;
pNode->next=pClonedNode->next;
pNode=pNode->next;
}
return pClonedHead;
}
Node* copyRandomList(Node* head) {
CloneNodes(head);
ConnectRandom(head);
return ReconnectNodes(head);
}
};