问题描述
请实现 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]]
示例 2:
输入:head = [[1,1],[2,1]]
输出:[[1,1],[2,1]]
问题分析
对于一个简单链表来说,如果我们需要复制的话,需要把每个节点都进行复制然后逐个的链接起来,这样看起来就比较的简单,代码实现如下:
public Node copy(Node head){
if(head == null){
return null;
}
Node cur = head.next;
Node newHead = new Node(head.val);
Node curNew = newHead;
while(cur != null){
Node node = new Node(cur.val);
curNew.next = node;
curNew = node;
cur = cur.next;
}
return newHead;
}
但是这里涉及一个问题,就是还存在一个随机的节点指针指向的节点是随机的,这个需要我们处理,那么怎么才能够完成了,上面的代码逻辑是逐个创建并链接到一起的,但是随机节点不一定创建出来了,针对这个问题我们可能把创建好的新节点链接在已有的节点后面,然后再逐个遍历新节点的随机节点就是前面节点对应随机节点的下一个节点,最后将copy的新节点从原先的链表中提取出来;
- 复制新节点,并放在原有节点的后面;
- 遍历节点,并复制新节点的随机节点,该节点为原有节点的随机节点的下一个节点;
- 提取复制节点列表,保持原有列表不变;
代码实现
class Solution {
public Node copyRandomList(Node head) {
if(head == null){
return head;
}
Node cur = head;
while(cur != null){
Node newNode = new Node(cur.val);
newNode.next = cur.next;
cur.next = newNode;
cur = newNode.next;
}
cur = head;
//copy random
while(cur != null && cur.next != null){
if(cur.random != null){
cur.next.random = cur.random.next;
}
cur = cur.next.next;
}
Node newHead = head.next;
Node copy = newHead;
head.next = head.next.next;
if(copy.next != null){
cur = copy.next;
}
while(cur != null && cur.next != null){
copy.next = cur.next;
cur.next = cur.next.next;
copy = copy.next;
if(copy != null){
cur = copy.next;
}
}
return newHead;
}
}