#include <vector> #include <iostream> #include <algorithm> #include <string> #include <map> #include <set> using namespace std; vector<bool> koloryPocz, koloryKon; vector<vector<int>> sasiedzi; int n; bool wszystkieRowne, takieSame; void wczytaj() { ios_base::sync_with_stdio(false); string strPocz, strKon; int i, a, b; wszystkieRowne = true; takieSame = true; cin >> n >> strPocz >> strKon; koloryPocz.resize(n); koloryKon.resize(n); sasiedzi.clear(); sasiedzi.resize(n); for (i = 0; i < n; i++) { koloryPocz[i] = strPocz[i] == '1'; koloryKon[i] = strKon[i] == '1'; wszystkieRowne = wszystkieRowne && (koloryPocz[i] == koloryPocz[0]); takieSame = takieSame && (koloryPocz[i] == koloryKon[i]); } for (i = 1; i < n; i++) { cin >> a >> b; a--; b--; sasiedzi[a].push_back(b); sasiedzi[b].push_back(a); } } int policzZmiany(int pocz, int kon, vector<bool>& kolory) { int poprz = pocz; int nast = sasiedzi[poprz][0]; int zmiany = kolory[poprz] != kolory[nast] ? 1 : 0; vector<bool> odwiedzone(n, false); odwiedzone[poprz] = odwiedzone[nast] = true; while (nast != kon) { // cerr << poprz << ' ' << nast; // cerr << " " << sasiedzi[nast][0] << " " << sasiedzi[nast][1]; // cerr << " " << odwiedzone[sasiedzi[nast][0]] << " " << odwiedzone[sasiedzi[nast][1]]; // cerr << endl; int curr = nast; nast = sasiedzi[curr][0]; if (odwiedzone[nast]) nast = sasiedzi[curr][1]; poprz = curr; odwiedzone[nast] = true; zmiany += kolory[poprz] != kolory[nast] ? 1 : 0; }; // cerr << "------------" << endl; return zmiany; } bool rozwiaz() { vector<int> konce; for (int i = 0; i < n; i++) { if (sasiedzi[i].size() == 1) { konce.push_back(i); if (konce.size() == 3) { return true; } } } int zmianyPocz = policzZmiany(konce[0], konce[1], koloryPocz); int zmianyKon = policzZmiany(konce[0], konce[1], koloryKon); if (zmianyKon < zmianyPocz) { return true; } else if (zmianyKon > zmianyPocz) { return false; } else { return koloryPocz[konce[0]] == koloryKon[konce[0]]; } } int main() { int t; cin >> t; while (t--) { wczytaj(); bool daSie; if (takieSame) { daSie = true; } else if (wszystkieRowne) { daSie = false; } else { daSie = rozwiaz(); } cout << (daSie ? "TAK" : "NIE") << endl; } 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 | #include <vector> #include <iostream> #include <algorithm> #include <string> #include <map> #include <set> using namespace std; vector<bool> koloryPocz, koloryKon; vector<vector<int>> sasiedzi; int n; bool wszystkieRowne, takieSame; void wczytaj() { ios_base::sync_with_stdio(false); string strPocz, strKon; int i, a, b; wszystkieRowne = true; takieSame = true; cin >> n >> strPocz >> strKon; koloryPocz.resize(n); koloryKon.resize(n); sasiedzi.clear(); sasiedzi.resize(n); for (i = 0; i < n; i++) { koloryPocz[i] = strPocz[i] == '1'; koloryKon[i] = strKon[i] == '1'; wszystkieRowne = wszystkieRowne && (koloryPocz[i] == koloryPocz[0]); takieSame = takieSame && (koloryPocz[i] == koloryKon[i]); } for (i = 1; i < n; i++) { cin >> a >> b; a--; b--; sasiedzi[a].push_back(b); sasiedzi[b].push_back(a); } } int policzZmiany(int pocz, int kon, vector<bool>& kolory) { int poprz = pocz; int nast = sasiedzi[poprz][0]; int zmiany = kolory[poprz] != kolory[nast] ? 1 : 0; vector<bool> odwiedzone(n, false); odwiedzone[poprz] = odwiedzone[nast] = true; while (nast != kon) { // cerr << poprz << ' ' << nast; // cerr << " " << sasiedzi[nast][0] << " " << sasiedzi[nast][1]; // cerr << " " << odwiedzone[sasiedzi[nast][0]] << " " << odwiedzone[sasiedzi[nast][1]]; // cerr << endl; int curr = nast; nast = sasiedzi[curr][0]; if (odwiedzone[nast]) nast = sasiedzi[curr][1]; poprz = curr; odwiedzone[nast] = true; zmiany += kolory[poprz] != kolory[nast] ? 1 : 0; }; // cerr << "------------" << endl; return zmiany; } bool rozwiaz() { vector<int> konce; for (int i = 0; i < n; i++) { if (sasiedzi[i].size() == 1) { konce.push_back(i); if (konce.size() == 3) { return true; } } } int zmianyPocz = policzZmiany(konce[0], konce[1], koloryPocz); int zmianyKon = policzZmiany(konce[0], konce[1], koloryKon); if (zmianyKon < zmianyPocz) { return true; } else if (zmianyKon > zmianyPocz) { return false; } else { return koloryPocz[konce[0]] == koloryKon[konce[0]]; } } int main() { int t; cin >> t; while (t--) { wczytaj(); bool daSie; if (takieSame) { daSie = true; } else if (wszystkieRowne) { daSie = false; } else { daSie = rozwiaz(); } cout << (daSie ? "TAK" : "NIE") << endl; } return 0; } |