剑指offer
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
思想:
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)
{
ClonelistNode(pHead);
CloneRandom(pHead);
return reconnectNodes(pHead);
}
void ClonelistNode(RandomListNode* pHead)
{
RandomListNode *p = pHead;
while(p != NULL)
{
RandomListNode *cloneNode = new RandomListNode(p->label);
cloneNode->next = p->next;
cloneNode->random = NULL;
p->next = cloneNode;
p = cloneNode->next;
}
}
void CloneRandom(RandomListNode* pHead)
{
RandomListNode *p = pHead;
while(p != NULL)
{
RandomListNode *cloneNode = p->next;
if(p->random != NULL)
{
cloneNode->random = p->random->next;
}
p = cloneNode->next;
}
}
//拆分链表
RandomListNode* reconnectNodes(RandomListNode* pHead){
if(pHead == NULL)
return NULL;
RandomListNode* p=pHead;
RandomListNode *cloneHead=p->next;
RandomListNode *cloneNode=cloneHead;
p->next=cloneNode->next;
p=p->next;
while(p!=NULL){
cloneNode->next=p->next;
cloneNode=cloneNode->next;
p->next=cloneNode->next;
p=p->next;
}
return cloneHead;
}
};