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) 来获得相关源码。