#include <bits/stdc++.h>
using namespace std;
vector<int> g[100005];
vector<int> ord;
void get_ord(int v, int par) {
ord.push_back(v);
for (auto u : g[v]) {
if (par != u)
get_ord(u, v);
}
}
int x;
int check_alternate(int v, int par, string &s) {
int alt;
if (s[v - 1] == '0') alt = 1;
else alt = 2;
for (auto u : g[v]) {
if (u != par) {
int alt_son = check_alternate(u, v, s);
if (alt_son == 0) return 0;
else if (alt_son == alt) return 0;
}
}
return alt;
}
string solve() {
int n, cnt[2] = {0, 0}, a, b;
string src, dst;
cin >> n >> src >> dst;
for (int i = 1; i <= n; i++) {
g[i].clear();
}
for (int i = 1; i < n; i++) {
cin >> a >> b;
g[a].push_back(b);
g[b].push_back(a);
}
if (src == dst) return "TAK";
int big = 0, idx = 1;
for (int i = 1; i <= n; i++) {
cnt[src[i-1] - '0']++;
if (g[i].size() > 2) {
big++;
idx = i;
}
}
int biale = cnt[0], czarne = cnt[1];
if (cnt[0] == n || cnt[0] == 0) {
return "NIE";
}
ord.clear();
for (int i = 1; i <= n; i++) {
if (g[i].size() == 1) {
get_ord(i, 0);
break;
}
}
if (big == 0) {
vector<int> blok_src, blok_dst;
for (auto i : ord) {
if (blok_src.empty() || blok_src.back() != src[i-1]) blok_src.push_back(src[i-1]);
if (blok_dst.empty() || blok_dst.back() != dst[i-1]) blok_dst.push_back(dst[i-1]);
}
// cout << "jojo" << endl;
// for (auto i : blok_src) cout << i << " ";cout << "\n";
// for (auto i : blok_dst) cout << i << " " ;cout << "\n";
if (blok_dst.size() > blok_src.size()) return "NIE";
if (blok_dst.size() == blok_src.size()) return (blok_dst == blok_src ? "TAK" : "NIE");
return "TAK";
}
int x = check_alternate(ord[0], 0, src);
int y = check_alternate(ord[0], 0, dst);
if (x == y || y == 0) return "TAK";
return "NIE";
}
int32_t main() {
ios_base::sync_with_stdio(0);
int tests;
cin >> tests;
while (tests--) cout << solve() << "\n";
}
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 | #include <bits/stdc++.h> using namespace std; vector<int> g[100005]; vector<int> ord; void get_ord(int v, int par) { ord.push_back(v); for (auto u : g[v]) { if (par != u) get_ord(u, v); } } int x; int check_alternate(int v, int par, string &s) { int alt; if (s[v - 1] == '0') alt = 1; else alt = 2; for (auto u : g[v]) { if (u != par) { int alt_son = check_alternate(u, v, s); if (alt_son == 0) return 0; else if (alt_son == alt) return 0; } } return alt; } string solve() { int n, cnt[2] = {0, 0}, a, b; string src, dst; cin >> n >> src >> dst; for (int i = 1; i <= n; i++) { g[i].clear(); } for (int i = 1; i < n; i++) { cin >> a >> b; g[a].push_back(b); g[b].push_back(a); } if (src == dst) return "TAK"; int big = 0, idx = 1; for (int i = 1; i <= n; i++) { cnt[src[i-1] - '0']++; if (g[i].size() > 2) { big++; idx = i; } } int biale = cnt[0], czarne = cnt[1]; if (cnt[0] == n || cnt[0] == 0) { return "NIE"; } ord.clear(); for (int i = 1; i <= n; i++) { if (g[i].size() == 1) { get_ord(i, 0); break; } } if (big == 0) { vector<int> blok_src, blok_dst; for (auto i : ord) { if (blok_src.empty() || blok_src.back() != src[i-1]) blok_src.push_back(src[i-1]); if (blok_dst.empty() || blok_dst.back() != dst[i-1]) blok_dst.push_back(dst[i-1]); } // cout << "jojo" << endl; // for (auto i : blok_src) cout << i << " ";cout << "\n"; // for (auto i : blok_dst) cout << i << " " ;cout << "\n"; if (blok_dst.size() > blok_src.size()) return "NIE"; if (blok_dst.size() == blok_src.size()) return (blok_dst == blok_src ? "TAK" : "NIE"); return "TAK"; } int x = check_alternate(ord[0], 0, src); int y = check_alternate(ord[0], 0, dst); if (x == y || y == 0) return "TAK"; return "NIE"; } int32_t main() { ios_base::sync_with_stdio(0); int tests; cin >> tests; while (tests--) cout << solve() << "\n"; } |
English