36、复杂链表的复制

输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)

题目链接:https://www.nowcoder.com/practice/f836b2c43afc4b35ad6adc41ec941dba

import java.util.Map;
import java.util.HashMap;
/*
public class RandomListNode {
    int label;
    RandomListNode next = null;
    RandomListNode random = null;

    RandomListNode(int label) {
        this.label = label;
    }
}
*/
public class Solution {
    public RandomListNode Clone(RandomListNode pHead)
    {
        if(pHead==null) return null;
        RandomListNode head = pHead;
        RandomListNode lastNode = null;
        RandomListNode coppiedNode;
        // 用一个map来保存节点head和它对应的复制节点coppiedNode的映射关系,
        // 用于设置 coppiedNode 的 random 时可以根据 head.random 快速得到对应的节点 coppiedNode.random
        Map map = 
            new HashMap();

        // 首先复制链表节点,并将复制节点的next安排起来
        while(head!=null){
            coppiedNode = new RandomListNode(head.label);
            map.put(head, coppiedNode);
            if(lastNode!=null){
                lastNode.next = coppiedNode;
            }
            lastNode = coppiedNode;
            head = head.next;
        }

        // 然后利用map中保存的节点映射关系得到对应节点的random节点
        head = pHead;
        while(head!=null){
            coppiedNode = map.get(head);
            coppiedNode.random = map.get(head.random);
            head = head.next;
        }
        
        return map.get(pHead);
        
    }
}

你可能感兴趣的:(36、复杂链表的复制)