剑指offer打卡|复杂链表的复制

题目描述

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

 

代码

本题有两种方法。

一、哈希表

import java.util.*;
public class Solution {
    //方法一：哈希表
    public RandomListNode Clone(RandomListNode pHead)
    {
        HashMap<RandomListNode,RandomListNode> map=new HashMap<>();
        RandomListNode cur=pHead;
        //构建哈希表，原结点与复制结点一一对应
        while(cur!=null){
            map.put(cur,new RandomListNode(cur.label));
            cur=cur.next;
        }
        cur=pHead;
        while(cur!=null){
            map.get(cur).next=map.get(cur.next);
            map.get(cur).random=map.get(cur.random);
            cur=cur.next;
        }
        return map.get(pHead);   
    }
}

二、把复制的结点放在原结点next的位置，以这样的方式代替哈希表中的对应关系

public RandomListNode Clone(RandomListNode pHead)
    {
        if(pHead==null){
            return null;
        }
        RandomListNode cur=pHead;
        RandomListNode next=null;
        //插入复制的结点
        while(cur!=null){
            next=cur.next;//2
            cur.next=new RandomListNode(cur.label);//1->1'
            cur.next.next=next;//1'->2
            cur=next;//指针从1移动到2，进行新一轮的复制
        }
        //设置复制结点的random指向
        cur=pHead;
        RandomListNode curcopy=null;
        while(cur!=null){
            next=cur.next.next;
            curcopy=cur.next;
            curcopy.random=cur.random!=null? cur.random.next:null;
            cur=next;
        }
        RandomListNode result=pHead.next;
        cur=pHead;
        //分离两个链表
        while(cur!=null){
            next=cur.next.next;
            curcopy=cur.next;
            cur.next=next;
            curcopy.next=next!=null? next.next:null;
            cur=next;
        }
        return result;
    }