-
Notifications
You must be signed in to change notification settings - Fork 26
/
dfs.cpp
82 lines (68 loc) · 1.32 KB
/
dfs.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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
#pragma GCC optimize("Ofast")
#include <algorithm>
#include <bitset>
#include <deque>
#include <iostream>
#include <iterator>
#include <string>
#include <map>
#include <queue>
#include <set>
#include <stack>
#include <vector>
#include <unordered_map>
#include <unordered_set>
using namespace std;
void abhisheknaiidu()
{
ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
#ifndef ONLINE_JUDGE
freopen("input.txt", "r", stdin);
freopen("output.txt", "w", stdout);
#endif
}
template <typename T>
class Graph {
unordered_map <T, vector<T> > m;
public:
void addEdge(int u, int v) {
m[u].push_back(v);
m[v].push_back(u);
}
void dfs_helper(T src, unordered_map<T, bool> &visited) {
cout << src << " ";
visited[src] = true;
// visit all neightbours of src
for(auto nbr: m[src]) {
if(!visited[nbr]){
dfs_helper(nbr, visited);
}
}
}
void dfs(T src) {
unordered_map <T, bool> visited;
for( auto p: m) {
T node = p.first;
visited[node] = false;
}
dfs_helper(src, visited);
}
// void print() {
// for(int i=0; i<V; i++) {
// cout << i << "->";
// for( auto x: adjList[i]) {
// cout << x << ",";
// }
// cout<< endl;
// }
// }
};
int main(int argc, char* argv[]) {
abhisheknaiidu();
Graph <int> g;
g.addEdge(0,1);
g.addEdge(0,2);
g.addEdge(2,3);
g.dfs(0);
return 0;
}