[和小菜鸡一起刷题(python)] LeetCode 138. 复制带随机指针的链表（Copy List with Random Pointer）

原题

给定一个链表，每个节点包含一个额外增加的随机指针，该指针可以指向链表中的任何节点或空节点。

要求返回这个链表的深度拷贝。

思路

先对链表进行一次遍历，在遍历过程中复制每一个链表节点。同时考虑到要复制随机指针，在遍历的同时创建原始链表节点和复制后的链表节点的关系，此处python中用dict，类似哈希的方式存储两者的对应关系。

再对复制后的链表进行一次遍历，遍历过程中根据存下的对应关系修改复制节点中随机指针的值。

代码

# Definition for singly-linked list with a random pointer.

# class RandomListNode(object):

#    def __init__(self, x):

#        self.label = x

#        self.next = None

#        self.random = None

class Solution(object):

    def copyRandomList(self, head):

        """

        :type head: RandomListNode

        :rtype: RandomListNode

        """

        dummy = RandomListNode(0)

        ori = head

        cp = dummy

        ori_cp_dict = {}     

        while(ori):

            cp.next = RandomListNode(ori.label)

            cp.next.random = ori.random

            ori_cp_dict[ori] = cp.next

            ori = ori.next

            cp = cp.next

        cp = dummy.next

        while(cp):

            if cp.random:

                cp.random = ori_cp_dict[cp.random]

            cp = cp.next 

        return dummy.next