题目:
请实现 copyRandomList 函数,复制一个复杂链表。在复杂链表中,每个节点除了有一个 next 指针指向下一个节点,还有一个 random 指针指向链表中的任意节点或者 null。
示例 1:
输入:head = [[7,null],[13,0],[11,4],[10,2],[1,0]]
输出:[[7,null],[13,0],[11,4],[10,2],[1,0]]
题解:假设原结点为A-B-C-D-E
复制后的结点为A’-B’-C’-D’-E’
一共分为三个步骤
1、复制每个结点,而且让每个复制后的结点接在被复制的节点后面:
A-A’-B-B’-C-C’-D-D’-E-E’
2、复制每个A’的random结点信息,让它通向相应的D’
3、将复制的链表从长链表中剥离出来
/*
// 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:
Node* copyRandomList(Node* head) {
//1.1判断null表
if (NULL == head) return head;
//1.2定义一个复制表头
Node* cur = head;
while (cur != NULL) {
//1.3拷贝一个copynode的值为A',假设原链表的值为A
Node* copyNode = new Node(cur->val);
//1.4将这个copynode插入到A后面
copyNode->next = cur->next;
cur->next = copyNode;
//1.5更新这个被拷贝的节点,即为A->A'->B,cur就是被拷贝的结点指针
cur = cur->next->next;
}
//2.1现在得到一个拷贝后的链表,下面就是将这个链表的random同样复制一次
cur = head;//找到头部
while(cur != NULL) {
if (cur->random != NULL) {
//2.2将cur(A)的random指针(D)的拷贝指针(D') 交给 cur.next(A')指针
cur->next->random = cur->random->next;
}
//2.3更新cur(A)的数据,也就是把它更新成B,也就是后两位的next
cur = cur->next->next;
}
//3.1现在得到一个链表,包含两份完全相同的数据元素,A-A'-B-B'-C-C'-D-D'-E-E'
//现在把他们拆开,只留下复制后的那个链表 A-B-C-D-E和A'-B'-C'-D'-E'
cur = head;//3.2先找到头结点
Node* copyHead = head->next;
Node* copyCur = head->next;
while (cur != NULL) {
//3.3 cur:切断A-A'
cur->next = cur->next->next;
//3.4 更新cur,变成B----
cur = cur->next;
//3.5 copycur:切断A'-B
if (copyCur->next != NULL) {
copyCur->next = copyCur->next->next;
//3.6 更新curcopy,变成B'------
copyCur = copyCur->next;
}
}
return copyHead;
}
};