复杂链表的复制

复杂链表的复制

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


用一个哈希表表示映射关系:键是原节点,值是复制的节点。

整体算法流程是:

  • 第一次遍历,复制每个节点和 next 指针,并且保存“原节点-复制节点”的映射关系
  • 第二次遍历,通过哈希表获得节点对应的复制节点,更新 random 指针

/*
public class RandomListNode {
    int label;
    RandomListNode next = null;
    RandomListNode random = null;

    RandomListNode(int label) {
        this.label = label;
    }
}
*/
import java.util.*;
public class Solution {
    public RandomListNode Clone(RandomListNode pHead)
    {
        if(pHead==null)
            return null;
        RandomListNode newHead=new RandomListNode(pHead.label);
        RandomListNode saveOld=pHead;
        RandomListNode saveNew=newHead;
        HashMap<RandomListNode,RandomListNode>mapping=new HashMap<RandomListNode,RandomListNode>();
        mapping.put(pHead,newHead);
         while(pHead.next!=null){
            newHead.next=new RandomListNode(pHead.next.label);
            mapping.put(pHead.next,newHead.next);
            pHead=pHead.next;
            newHead=newHead.next;
        }
        pHead=saveOld;
        newHead=saveNew;
        while(pHead!=null){
            newHead.random=mapping.get(pHead.random);
            pHead=pHead.next;
            newHead=newHead.next;
        }
        return saveNew;
    }
}

要点:

  1. map的插入是put,根据key找value是get
  2. 注意判断pHead是否为null
posted @ 2020-03-05 11:56  别再闹了  阅读(39)  评论(0)    收藏  举报