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);
    }
}
posted @ 2025-05-27 09:50  回忆、少年  阅读(10)  评论(0)    收藏  举报