题目一来,思路的话,主要是想拿个数组或者vector来存这些东西,存在什么问题呢?
- 数组的话,想要知道自身在数组中的位置是可以做到的吗?
针对于这个问题,感觉可以用arraylist,或者hashmap来存。因为问题的关键在于,某个节点random给i另外一个节点的时候,你得找得到对应的节点才行,也就是说,你一定得给原链表一个标识,比如你用arraylist来存原本的链表,那么第四个节点random给第一个节点,你得能表现出4->1这个信息才行。所以需要一个映射。说到映射就想到用字典,就是用一个<old_node,new_node>的一个字典。所以本题我还是倾向于用map来做,即哈希表来做。
做法流程如下:
- 新建一个map作为映射
- 对链表进行一次遍历,对每个旧节点进行复制,得到的新节点,进行map的装入
- 对老链表进行遍历,其中弄好映射关系。
/*
// Definition for a Node.
class Node {
int val;
Node next;
Node random;
public Node(int val) {
this.val = val;
this.next = null;
this.random = null;
}
}
*/
class Solution {
public Node copyRandomList(Node head) {
// 想法是 那个数组来存链表的位置,然后再随机进行链接 //数组本来也行,但是呢,数组做不到索引这个事情,即让数组的元素返回自己所在的位置是很难返回出来的。所以要用到字典,也就是常见所说的哈希表。
Map<Node,Node> map = new HashMap<>();
Node cur = head;
if(cur==null){
return null;
}
while(cur!=null){
map.put(cur,new Node(cur.val));
cur = cur.next;
}
cur = head;
while(cur!=null){
map.get(cur).next = map.get(cur.next);
map.get(cur).random = map.get(cur.random);
cur = cur.next;
}
return map.get(head);
}
}