题目链接:
https://leetcode-cn.com/problems/fu-za-lian-biao-de-fu-zhi-lcof/
题意:
请实现 copyRandomList 函数,复制一个复杂链表。在复杂链表中,每个节点除了有一个 next 指针指向下一个节点,还有一个 random 指针指向链表中的任意节点或者 null。
题解:
拼接+拆分:
需要三次遍历。
- 第一次遍历:将链表上所有的节点的val以及next复制一遍。并按照
原->新->原->新->..
进行拼接。 - 第二次遍历:由第一遍的最终结构,可以得到
新.random = 原.random.next
,所以在第二遍遍历中将新.random
给全部找出来。 - 第三次遍历:将所有的
新
给拆分出来,形成一个新的链表。
代码:
"""
# Definition for a Node.
class Node:
def __init__(self, x: int, next: 'Node' = None, random: 'Node' = None):
self.val = int(x)
self.next = next
self.random = random
"""
class Solution:
def copyRandomList(self, head: 'Node') -> 'Node':
if not head :
return None
s1,s2,p1,p2= head,None,Node(0),None
while s1!=None:
p1.val =s1.val
s2=s1
s1=s1.next
s2.next=p1
p1.next=Node(0)
p2=p1
p1=p1.next
p2.next =s1
s1,p1= head,head.next
while s1!=None:
p1.random =s1.random.next if s1.random else None
if p1.next!=None:
s1,p1=s1.next.next,p1.next.next
else:
s1=None
ret,p1= head.next,head.next;
while p1!=None:
if p1.next:
p1.next = p1.next.next
p1=p1.next
else:
p1=None
return ret