算法刷题之路之链表初探(四)
今天来学习的算法题是leecode138随机链表的赋值,是一道简单的入门题,话不多说!直接上!
条件
代码
/*
// 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) {
if(head == null){
return null;
}
Node cur = head;
HashMap<Node,Node> map = new HashMap<>();
while(cur!=null){
//建立新链表和旧链表的对应关系(1对1)
map.put(cur,new Node(cur.val));
cur = cur.next;
}
cur=head;
while(cur!=null){
//map.get(cur)是获取到新链表中的node节点。
//因为是一对以的对应关系,所以旧节点的next就等于新节点的next
//所以只需要拿到旧节点对应的next值赋值给新节点就完成了新节点的next属性的赋值
map.get(cur).next=map.get(cur.next);
//随机链表也是一样的道理
map.get(cur).random=map.get(cur.random);
cur=cur.next;
}
return map.get(head);
}
}