题目描述:
请实现 copyRandomList 函数,复制一个复杂链表。在复杂链表中,每个节点除了有一个 next 指针指向下一个节点,还有一个 random 指针指向链表中的任意节点或者 null。
及
解题思路:
1. 使用哈希表,存储每个结点复制后的新结点;
2. 在得到新结点与旧结点的映射关系后,通过某种方式 将新结点按照旧结点
的关系进行连线。
3. 连线方式:遍历到原始链表的当前结点cur, 从 mp 中取出 cur 结点映射
的结点 newCur, 我们需要做的就是 将newCur 按照 cur 在原始链表中
的关系连线, 因为是要对 新结点 进行连线, 因此所有的结点都必须 从
map中取对应的新结点
newCur.next 对应的 next 在map中的映射为 map.get(cur.next);
newCur.random 对应的 random 在 map中映射为
map.get(cur.random);
4. 返回头结点, map.get(head);
class Solution {
public Node copyRandomList(Node head) {
if(head == null) return head;
Node cur = head;
Map<Node,Node> mp = new HashMap();
while(cur != null){
mp.put(cur,new Node(cur.val));
cur = cur.next;
}
cur = head;
while(cur != null){
mp.get(cur).next = mp.get(cur.next);
mp.get(cur).random = mp.get(cur.random);
cur = cur.next;
}
return mp.get(head);
}
}