#include <bits/stdc++.h> using namespace std; int main() { ios::sync_with_stdio(false); int n, m; cin >> n >> m; vector<char> topo(n+1); vector<unordered_set<int>> strong_out(n+1), strong_in(n+1), weak_out(n+1), weak_in(n+1); vector<vector<bool>> transistive_inferiors(n+1, vector<bool>(n+1)); vector<int> parents(n+1); while (m--) { int a, b; string s; cin >> a >> b >> s; if (s == "N") { weak_out[a].insert(b); weak_in[b].insert(a); } else { strong_out[b].insert(a); strong_in[a].insert(b); } } int director = 0; for (int i = 1; i <= n; ++i) { if (weak_in[i].empty() && strong_in[i].empty()) { director = i; break; } } //cerr << "Director = " << director << endl; if (director == 0) { cout << "NIE" << endl; return 0; } for (int i = 1; i <= n; ++i) { if (i == director) continue; strong_in[i].insert(director); strong_out[director].insert(i); } vector<int> parent(n+1); vector<bool> processed(n+1); for (int z = 0; z<n; ++z) { //cerr << "Next round." << endl; vector<int> component(n+1); vector<vector<int>> components(1); int component_id = 0; for (int i = 1; i <= n; ++i) { if (processed[i] || component[i]!=0) continue; component_id++; components.emplace_back(); queue<int> q; q.push(i); while(!q.empty()) { int a = q.front(); q.pop(); component[a] = component_id; components[component_id].push_back(a); for (int b : strong_in[a]) { if (!processed[b] && component[b]==0) { q.push(b); component[b] = component_id; } } for (int b : strong_out[a]) { if (!processed[b] && component[b]==0) { q.push(b); component[b] = component_id; } } } } for (int c = 1; c <= component_id; ++c) { //cerr << "Component " << c << endl; int director = 0; for (int a : components[c]) { //cerr << a << ", "; bool empty = true; for (int x : strong_in[a]) if (component[x] == component[a]) empty = false; for (int x : weak_in[a]) if (component[x] == component[a]) empty = false; if (empty) director = a; } //cerr << endl; //cerr << "Director = " << director << endl; if (director == 0) { cout << "NIE" << endl; return 0; } for (int b : weak_out[director]) weak_in[b].erase(director); for (int b : strong_out[director]) strong_in[b].erase(director); weak_out[director].clear(); strong_out[director].clear(); processed[director] = true; for (int a : components[c]) { if (a!= director) parent[a] = director; } } } for (int i = 1; i<=n; ++i) cout << parent[i] << endl; //cout << "TAK" << endl; }
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 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 | #include <bits/stdc++.h> using namespace std; int main() { ios::sync_with_stdio(false); int n, m; cin >> n >> m; vector<char> topo(n+1); vector<unordered_set<int>> strong_out(n+1), strong_in(n+1), weak_out(n+1), weak_in(n+1); vector<vector<bool>> transistive_inferiors(n+1, vector<bool>(n+1)); vector<int> parents(n+1); while (m--) { int a, b; string s; cin >> a >> b >> s; if (s == "N") { weak_out[a].insert(b); weak_in[b].insert(a); } else { strong_out[b].insert(a); strong_in[a].insert(b); } } int director = 0; for (int i = 1; i <= n; ++i) { if (weak_in[i].empty() && strong_in[i].empty()) { director = i; break; } } //cerr << "Director = " << director << endl; if (director == 0) { cout << "NIE" << endl; return 0; } for (int i = 1; i <= n; ++i) { if (i == director) continue; strong_in[i].insert(director); strong_out[director].insert(i); } vector<int> parent(n+1); vector<bool> processed(n+1); for (int z = 0; z<n; ++z) { //cerr << "Next round." << endl; vector<int> component(n+1); vector<vector<int>> components(1); int component_id = 0; for (int i = 1; i <= n; ++i) { if (processed[i] || component[i]!=0) continue; component_id++; components.emplace_back(); queue<int> q; q.push(i); while(!q.empty()) { int a = q.front(); q.pop(); component[a] = component_id; components[component_id].push_back(a); for (int b : strong_in[a]) { if (!processed[b] && component[b]==0) { q.push(b); component[b] = component_id; } } for (int b : strong_out[a]) { if (!processed[b] && component[b]==0) { q.push(b); component[b] = component_id; } } } } for (int c = 1; c <= component_id; ++c) { //cerr << "Component " << c << endl; int director = 0; for (int a : components[c]) { //cerr << a << ", "; bool empty = true; for (int x : strong_in[a]) if (component[x] == component[a]) empty = false; for (int x : weak_in[a]) if (component[x] == component[a]) empty = false; if (empty) director = a; } //cerr << endl; //cerr << "Director = " << director << endl; if (director == 0) { cout << "NIE" << endl; return 0; } for (int b : weak_out[director]) weak_in[b].erase(director); for (int b : strong_out[director]) strong_in[b].erase(director); weak_out[director].clear(); strong_out[director].clear(); processed[director] = true; for (int a : components[c]) { if (a!= director) parent[a] = director; } } } for (int i = 1; i<=n; ++i) cout << parent[i] << endl; //cout << "TAK" << endl; } |