我们很容易可以想到HashMap的做法,但是O(n)的空间复杂度,对于链表的题而言,明显是一个减分项,因此,可以采用下面的做法:
- 复制一个新的节点在原有节点之后,如 1 -> 2 -> 3 -> null 复制完就是 1 -> 1 -> 2 -> 2 -> 3 - > 3 -> null
- 从头开始遍历链表,通过 cur.next.random = cur.random.next 可以将复制节点的随机指针串起来,当然需要判断 cur.random 是否存在
- 将复制完的链表一分为二。
- 同时也要注意将原来的链表还原。
class Solution {
public Node copyRandomList(Node head) {
if(head==null){
return head;
}
Node cur = head;
while(cur!=null){
Node temp = cur.next;
Node node = new Node(cur.val);
cur.next = node;
node.next = temp;
cur = temp;
}
cur = head;
while(cur!=null){
if(cur.random != null){
cur.next.random = cur.random.next;
}
cur = cur.next.next;
}
cur = head;
Node ret = cur.next;
Node tail = ret;
while(cur!=null){
Node temp = cur.next.next;
if(temp!=null){
tail.next = temp.next;
tail = temp.next;
}
cur.next = temp;
cur = temp;
}
return ret;
}
}