36.复杂链表的复刻
请实现一个函数可以复制一个复杂链表。
在复杂链表中,每个结点除了有一个指针指向下一个结点外,还有一个额外的指针指向链表中的任意结点或者null。
注意:
- 函数结束后原链表要与输入时保持一致。
数据范围:
链表长度 [0,500]。
代码:
/**
* Definition for singly-linked list with a random pointer.
* class ListNode {
* int val;
* ListNode next, random;
* ListNode(int x) { this.val = x; }
* };
*/
class Solution {
public ListNode copyRandomList(ListNode head) {
//如果当前链表头节点为null,直接返回null
if(head == null)return null;
//定义一个哈希表来记录节点和节点指针的映射
Map<ListNode,ListNode>mp = new HashMap<>();
//定义一个节点用来遍历链表
ListNode cur = head;
//如果当前节点不为空,就往后遍历链表
while(cur!=null){
//新建一个和原链表值相同的节点,并将原链表指针和该节点映射存入哈希表
mp.put(cur,new ListNode(cur.val));
//当前节点往后移动
cur = cur.next;
}
//将当前节点重新设置为头节点,以便进行新一轮的遍历
cur = head;
//如果当前节点不为空
while(cur!=null){
//定义一个节点,该节点从哈希表中获取
ListNode node = mp.get(cur);
//该节点的next节点也直接从哈希表中获取
node.next = mp.get(cur.next);
//该节点的random节点也从哈希表中直接获取
node.random = mp.get(cur.random);
//当前节点往后移动
cur = cur.next;
}
//返回头节点
return mp.get(head);
}
}

浙公网安备 33010602011771号