-
Notifications
You must be signed in to change notification settings - Fork 0
/
graphClone.cpp
52 lines (49 loc) · 1.2 KB
/
graphClone.cpp
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
/*
// Definition for a Node.
class Node {
public:
int val;
vector<Node*> neighbors;
Node() {
val = 0;
neighbors = vector<Node*>();
}
Node(int _val) {
val = _val;
neighbors = vector<Node*>();
}
Node(int _val, vector<Node*> _neighbors) {
val = _val;
neighbors = _neighbors;
}
};
*/
class Solution {
public:
Node* cloneGraph(Node* node) {
if(!node){
return NULL;
}
unordered_map<Node*, Node*> mp;
// referring copies
Node* copy = new Node(node->val, {});
mp[node] = copy;
// bfs traversal
queue<Node*> q;
q.push(node);
while(!q.empty()){
Node* curr = q.front();
q.pop();
for(Node* neighbour: curr -> neighbors){
if(mp.find(neighbour) == mp.end()){
// not found in the map then
q.push(neighbour);
mp[neighbour] = new Node(neighbour->val, {});
}
// filling the list for the current node
mp[curr] -> neighbors.push_back(mp[neighbour]);
}
}
return copy;
}
};