-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathdfs.cpp
75 lines (71 loc) · 1.92 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
// Created on 14-07-2019 11:10:00 by necronomicon
#include <bits/stdc++.h>
using namespace std;
#define MP make_pair
#define PB push_back
#define ARR_MAX (int)1e5 //Max array length
#define INF (int)1e9 //10^9
#define EPS 1e-9 //10^-9
#define MOD 1000000007 //10^9+7
#define PI 3.1415926535897932384626433832795
typedef long int int32;
typedef unsigned long int uint32;
typedef long long int int64;
typedef unsigned long long int uint64;
typedef pair<int, int> Pii;
typedef vector<int> Vi;
typedef vector<string> Vs;
typedef vector<Pii> VPii;
typedef vector<Vi> VVi;
typedef map<int,int> Mii;
typedef set<int> Si;
typedef multimap<int,int> MMii;
typedef multiset<int> MSi;
typedef unordered_map<int,int> UMii;
typedef unordered_set<int> USi;
typedef unordered_multimap<int,int> UMMii;
typedef unordered_multiset<int> UMSi;
typedef priority_queue<int> PQi;
typedef queue<int> Qi;
typedef deque<int> DQi;
// dfs using recursion
void dfsRecursion(VVi adj, vector<bool> &visited, int k){
if(visited[k]) return;
visited[k] = true;
cout << k << ' ';
for(int x: adj[k]) dfsRecursion(adj, visited, x);
}
// dfs using stack
void dfsStack(VVi adj, vector<bool> &visited, int k){
stack <int> S;
S.push(k);
while(!S.empty()){
int top = S.top();
S.pop();
if(visited[top]) continue;
else{
cout << top << ' ';
visited[top] = true;
for(int x: adj[top]) S.push(x);
}
}
}
int main (int argc, char const *argv[]) {
int N = 6; // number of nodes
VVi adj;
vector<bool> visited(N, false);
adj = {
{},
{2},
{0, 5},
{2},
{4, 3},
{}
};
for(int i=0; i<N; i++) visited[i]=false;
dfsRecursion(adj, visited, 1);
cout << endl;
for(int i=0; i<N; i++) visited[i]=false;
dfsStack(adj, visited, 1);
return EXIT_SUCCESS;
}