#include <iostream>
#include <vector>
#include <set>
#include <queue>
#include <algorithm>
using namespace std;
// Pomocnicza funkcja DSU find z kompresją ścieżki
int dsu_find(vector<int>& dsu, int i) {
if (dsu[i] == i) return i;
return dsu[i] = dsu_find(dsu, dsu[i]);
}
void solve() {
int n, m, k;
if (!(cin >> n >> m >> k)) return;
vector<int> a(n + 1);
vector<vector<int>> party_cities(k + 1);
vector<int> present_parties;
vector<bool> is_present(k + 1, false);
// Wczytywanie wyników w miastach
for (int i = 1; i <= n; ++i) {
cin >> a[i];
party_cities[a[i]].push_back(i);
if (!is_present[a[i]]) {
is_present[a[i]] = true;
present_parties.push_back(a[i]);
}
}
// Wczytywanie krawędzi (dróg)
vector<vector<int>> adj(n + 1);
for (int i = 0; i < m; ++i) {
int u, v;
cin >> u >> v;
adj[u].push_back(v);
adj[v].push_back(u);
}
vector<int> dsu(n + 1);
for (int i = 1; i <= n; ++i) dsu[i] = i;
// Początkowe łączenie miast tej samej partii
for (int u = 1; u <= n; ++u) {
for (int v : adj[u]) {
if (a[u] == a[v]) {
int r1 = dsu_find(dsu, u);
int r2 = dsu_find(dsu, v);
if (r1 != r2) dsu[r1] = r2;
}
}
}
// parties_in_root[r] przechowuje ID partii mających miasta w komponencie r
vector<set<int>> parties_in_root(n + 1);
vector<int> comp_count(k + 1, 0);
for (int p : present_parties) {
set<int> roots;
for (int city : party_cities[p]) {
roots.insert(dsu_find(dsu, city));
}
comp_count[p] = (int)roots.size();
for (int r : roots) {
parties_in_root[r].insert(p);
}
}
queue<int> q;
vector<bool> in_queue(k + 1, false);
for (int p : present_parties) {
if (comp_count[p] == 1) {
in_queue[p] = true;
q.push(p);
}
}
int processed_count = 0;
while (!q.empty()) {
int p = q.front();
q.pop();
processed_count++;
for (int v : party_cities[p]) {
for (int u : adj[v]) {
int r1 = dsu_find(dsu, v);
int r2 = dsu_find(dsu, u);
if (r1 == r2) continue;
// Łączenie mniejszego zbioru z większym (optymalizacja)
if (parties_in_root[r1].size() < parties_in_root[r2].size()) swap(r1, r2);
for (int party_id : parties_in_root[r2]) {
if (parties_in_root[r1].count(party_id)) {
// Jeśli obie składowe mają tę samą partię, liczba jej komponentów maleje
if (!in_queue[party_id]) {
comp_count[party_id]--;
if (comp_count[party_id] == 1) {
in_queue[party_id] = true;
q.push(party_id);
}
}
} else {
parties_in_root[r1].insert(party_id);
}
}
dsu[r2] = r1;
parties_in_root[r2].clear();
}
}
}
if (processed_count == (int)present_parties.size()) cout << "TAK\n";
else cout << "NIE\n";
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(NULL);
int t;
if (!(cin >> t)) return 0;
while (t--) {
solve();
}
return 0;
}
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 | #include <iostream> #include <vector> #include <set> #include <queue> #include <algorithm> using namespace std; // Pomocnicza funkcja DSU find z kompresją ścieżki int dsu_find(vector<int>& dsu, int i) { if (dsu[i] == i) return i; return dsu[i] = dsu_find(dsu, dsu[i]); } void solve() { int n, m, k; if (!(cin >> n >> m >> k)) return; vector<int> a(n + 1); vector<vector<int>> party_cities(k + 1); vector<int> present_parties; vector<bool> is_present(k + 1, false); // Wczytywanie wyników w miastach for (int i = 1; i <= n; ++i) { cin >> a[i]; party_cities[a[i]].push_back(i); if (!is_present[a[i]]) { is_present[a[i]] = true; present_parties.push_back(a[i]); } } // Wczytywanie krawędzi (dróg) vector<vector<int>> adj(n + 1); for (int i = 0; i < m; ++i) { int u, v; cin >> u >> v; adj[u].push_back(v); adj[v].push_back(u); } vector<int> dsu(n + 1); for (int i = 1; i <= n; ++i) dsu[i] = i; // Początkowe łączenie miast tej samej partii for (int u = 1; u <= n; ++u) { for (int v : adj[u]) { if (a[u] == a[v]) { int r1 = dsu_find(dsu, u); int r2 = dsu_find(dsu, v); if (r1 != r2) dsu[r1] = r2; } } } // parties_in_root[r] przechowuje ID partii mających miasta w komponencie r vector<set<int>> parties_in_root(n + 1); vector<int> comp_count(k + 1, 0); for (int p : present_parties) { set<int> roots; for (int city : party_cities[p]) { roots.insert(dsu_find(dsu, city)); } comp_count[p] = (int)roots.size(); for (int r : roots) { parties_in_root[r].insert(p); } } queue<int> q; vector<bool> in_queue(k + 1, false); for (int p : present_parties) { if (comp_count[p] == 1) { in_queue[p] = true; q.push(p); } } int processed_count = 0; while (!q.empty()) { int p = q.front(); q.pop(); processed_count++; for (int v : party_cities[p]) { for (int u : adj[v]) { int r1 = dsu_find(dsu, v); int r2 = dsu_find(dsu, u); if (r1 == r2) continue; // Łączenie mniejszego zbioru z większym (optymalizacja) if (parties_in_root[r1].size() < parties_in_root[r2].size()) swap(r1, r2); for (int party_id : parties_in_root[r2]) { if (parties_in_root[r1].count(party_id)) { // Jeśli obie składowe mają tę samą partię, liczba jej komponentów maleje if (!in_queue[party_id]) { comp_count[party_id]--; if (comp_count[party_id] == 1) { in_queue[party_id] = true; q.push(party_id); } } } else { parties_in_root[r1].insert(party_id); } } dsu[r2] = r1; parties_in_root[r2].clear(); } } } if (processed_count == (int)present_parties.size()) cout << "TAK\n"; else cout << "NIE\n"; } int main() { ios_base::sync_with_stdio(false); cin.tie(NULL); int t; if (!(cin >> t)) return 0; while (t--) { solve(); } return 0; } |
English