剑指offer——复杂链表的复制

Posted 2021-03-24 hit-joseph

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

解题思路：两边循环：

1：第一遍开两个数组，第一个存在每个节点，第二个存在每个节点值新开的节点。同时建立一个节点内存地址到index的映射id2index

2：遍历新开节点数组，按id2index补充next和random指针。

# class RandomListNode:
#     def __init__(self, x):
#         self.label = x
#         self.next = None
#         self.random = None
class Solution:
    # 返回 RandomListNode
    def Clone(self, pHead):
        # write code here
        if pHead==None:
            return None
        iter_array,array=[],[]
        id2index=dict()
        index=0
        #p=pHead
        while pHead:
            iter_array.append(pHead)
            array.append(RandomListNode(pHead.label))
            id2index[id(pHead)]=index
            index+=1
            pHead=pHead.next
        for p, res in zip(iter_array,array):
            if p.next:
                res.next=array[id2index[id(p.next)]]
            if p.random:
                res.random=array[id2index[id(p.random)]]
        return array[0]