《剑指offer》—— 35 复杂链表的复制

题目

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

思路1

• 利用哈希表，建立原链表和复制链表之间的映射，在复制链表上建立next和random关系，输出复制链表的头结点

代码1

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

    RandomListNode(int label) {
        this.label = label;
    }
}
*/
import java.util.HashMap;

public class Solution {
    public RandomListNode Clone(RandomListNode pHead)
    {
        RandomListNode temp = pHead;
        HashMap<RandomListNode, RandomListNode> map = new HashMap<>();
        //将复制的节点放在哈希表里
        while(temp != null){
            map.put(temp, new RandomListNode(temp.label));
            temp = temp.next;
        }
        
        //在哈希表的value之间建立next关系
        temp = pHead;
        while(temp != null){
            map.get(temp).next = map.get(temp.next);
            temp = temp.next;
        }
        
        //建立random
        temp = pHead;
        while(temp != null){
            map.get(temp).random = map.get(temp.random);
            temp = temp.next;
        }
        
        //value中的头结点
        RandomListNode clone = map.get(pHead);
        
        return clone;
    }
}

思路2

• 复制节点放在对应的源节点后面：A——A’——B——B’——
• 建立复制节点的random关系
• 拆开两个链表

代码2

    public RandomListNode Clone(RandomListNode pHead){
        if(pHead == null) return null;
        
        RandomListNode temp = pHead;
        //每个节点后面跟着它的复制节点
        while(temp != null){
            RandomListNode node = new RandomListNode(temp.label);
            node.next = temp.next;
            node.random = null;
            temp.next = node;
            temp = node.next;
        }
        
        //建立复制节点的random
        temp = pHead;
        while(temp != null){
            if(temp.random != null){
                temp.next.random = temp.random.next;  //要链接到复制的节点上temp.random.next
            }
            temp = temp.next.next;
        }
        
        //将两个链表拆开
        RandomListNode clone = pHead.next; //复制链表的头结点
        RandomListNode cTemp = clone; //复制链表的指针
        temp = pHead; //原链表的指针
        while(temp != null){
            temp.next = temp.next.next;  //修改原链表的next指向
            if(cTemp.next != null){
                cTemp.next = cTemp.next.next;
            }
            temp = temp.next;
            cTemp = cTemp.next;
        }
        return clone;
    }