LeetCode Copy List with Random Pointer

LeetCode解题之Copy List with Random Pointer

原题

一个链表中的每一个节点都有一个额外的随机指针，指向链表中的任意节点或空节点。对这个链表进行深拷贝。

注意点：

• 无

例子:

输入: 略

输出: 略

解题思路

比较容易想到的思路就是先不管新添加的指针，把链表当做普通链表先进行拷贝。因为新添加的指针可以指向任意的节点，所以可以将原节点和拷贝的节点建立映射关系，这样就可以找到任意节点对应的拷贝节点了。然后再次遍历原链表，对任意指针进行复制。

链表的拷贝其实可以看做两个步骤，一个是节点数据的拷贝，另一个是节点关系的拷贝。我们也可以先把所有的节点进行拷贝，并存入字典中。然后遍历链表并拷贝两个指针。因为任意指针可能指向空指针，所以在字典中添加一个空指针项。

AC源码

# 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 """
        if not head:
            return None
        visited = dict()
        node = head
        while node:
            visited[node] = RandomListNode(node.label)
            node = node.next

        visited[None] = None
        node = head
        while node:
            visited[node].next = visited[node.next]
            visited[node].random = visited[node.random]
            node = node.next
        return visited[head]


if __name__ == "__main__":
    None

欢迎查看我的Github (https://github.com/gavinfish/LeetCode-Python) 来获得相关源码。