forked from cp-algorithms/cp-algorithms
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_tarjan_scc.cpp
More file actions
66 lines (55 loc) · 1.95 KB
/
test_tarjan_scc.cpp
File metadata and controls
66 lines (55 loc) · 1.95 KB
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
#include <bits/stdc++.h>
using namespace std;
#include "tarjan_scc.h"
int main() {
// we only have a single test case for now
int n = 10;
vector<vector<int>> adj(n);
adj[0].push_back(1);
adj[0].push_back(7);
adj[1].push_back(1);
adj[1].push_back(2);
adj[2].push_back(1);
adj[2].push_back(5);
adj[3].push_back(2);
adj[3].push_back(4);
adj[4].push_back(9);
adj[5].push_back(3);
adj[5].push_back(6);
adj[5].push_back(9);
adj[6].push_back(2);
adj[7].push_back(0);
adj[7].push_back(6);
adj[7].push_back(8);
adj[8].push_back(6);
adj[8].push_back(9);
adj[9].push_back(4);
vector<vector<int>> components, adj_scc;
strongly_connected_components(adj, components, adj_scc);
auto sorted_components = components;
for (vector<int> &a : sorted_components)
sort(a.begin(), a.end());
sort(sorted_components.begin(), sorted_components.end(),
[](auto &l, auto &r) { return l[0] < r[0]; });
vector<int> root(n);
for (auto &comp : components) {
for (auto v : comp) {
root[v] = *comp.begin();
}
}
vector<int> minimal_element(n);
for (auto &comp : components) {
minimal_element[*comp.begin()] = *min_element(comp.begin(), comp.end());
}
for (vector<int> &a : adj_scc)
sort(a.begin(), a.end(), [minimal_element](auto &l, auto &r) { return minimal_element[l] < minimal_element[r]; });
assert(sorted_components.size() == 4);
assert(sorted_components[0] == std::vector<int>({0, 7}));
assert(sorted_components[1] == std::vector<int>({1, 2, 3, 5, 6}));
assert(sorted_components[2] == std::vector<int>({4, 9}));
assert(sorted_components[3] == std::vector<int>({8}));
assert(adj_scc[root[0]] == std::vector<int>({root[1], root[1], root[8]}));
assert(adj_scc[root[1]] == std::vector<int>({root[4], root[4]}));
assert(adj_scc[root[8]] == std::vector<int>({root[1], root[4]}));
return 0;
}