-
Notifications
You must be signed in to change notification settings - Fork 119
/
Copy pathClone Graph.py
38 lines (23 loc) · 987 Bytes
/
Clone Graph.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
def cloneGraph(self, node: 'Node') -> 'Node':
if node == None:
return None
new_node = Node(node.val, [])
visited = set()
q = [[node, new_node]]
visited.add(node.val)
adj_map = {}
adj_map[node] = new_node
while len(q) != 0:
curr = q.pop(0)
for n in curr[0].neighbors:
# if n.val not in visited:
if n not in adj_map and n is not None:
new = Node(n.val, [])
curr[1].neighbors.append(new)
adj_map[n] = new
else:
curr[1].neighbors.append(adj_map[n])
if n.val not in visited:
q.append([n, adj_map[n]])
visited.add(n.val)
return new_node