#include <cstdio>
#include <vector>
using namespace std;
typedef vector<vector<int>> E;
bool isPath(E& e) {
int cnt = 0;
for (int i=0; i<e.size(); ++i) {
if (e[i].size() < 2) ++cnt;
}
return cnt < 3;
}
int numFlips(E& e, vector<char>& a, int p = 0, int r = 0) {
int cnt = 0;
for (int q : e[p]) {
if (q == r) continue;
cnt += numFlips(e, a, q, p);
if (a[q] != a[p]) ++cnt;
}
return cnt;
}
void tak() {
printf("TAK\n");
}
void nie() {
printf("NIE\n");
}
void solve() {
int n; scanf("%d", &n);
vector<char> a(n), b(n);
for (int i=0; i<n; ++i) scanf(" %c", &a[i]);
for (int i=0; i<n; ++i) scanf(" %c", &b[i]);
E e(n);
for (int i=0; i<n-1; ++i) {
int x, y; scanf("%d %d", &x, &y); --x; --y;
e[x].push_back(y);
e[y].push_back(x);
}
int flipsA = numFlips(e, a);
int flipsB = numFlips(e, b);
if (isPath(e)) {
if (flipsA < flipsB) { nie(); return; }
if (flipsA > flipsB) { tak(); return; }
for (int i=0; i<n; ++i) {
if (e[i].size() < 2) {
if (a[i] == b[i]) tak(); else nie();
return;
}
}
}
bool same = true;
for (int i=0; i<n; ++i) if (a[i] != b[i]) same = false;
if (same) { tak(); return; }
if (flipsA < 1) { nie(); return; }
if (flipsB < n-1) tak(); else nie();
}
int main() {
int t; scanf("%d", &t);
for (int s=0; s<t; ++s) 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 | #include <cstdio> #include <vector> using namespace std; typedef vector<vector<int>> E; bool isPath(E& e) { int cnt = 0; for (int i=0; i<e.size(); ++i) { if (e[i].size() < 2) ++cnt; } return cnt < 3; } int numFlips(E& e, vector<char>& a, int p = 0, int r = 0) { int cnt = 0; for (int q : e[p]) { if (q == r) continue; cnt += numFlips(e, a, q, p); if (a[q] != a[p]) ++cnt; } return cnt; } void tak() { printf("TAK\n"); } void nie() { printf("NIE\n"); } void solve() { int n; scanf("%d", &n); vector<char> a(n), b(n); for (int i=0; i<n; ++i) scanf(" %c", &a[i]); for (int i=0; i<n; ++i) scanf(" %c", &b[i]); E e(n); for (int i=0; i<n-1; ++i) { int x, y; scanf("%d %d", &x, &y); --x; --y; e[x].push_back(y); e[y].push_back(x); } int flipsA = numFlips(e, a); int flipsB = numFlips(e, b); if (isPath(e)) { if (flipsA < flipsB) { nie(); return; } if (flipsA > flipsB) { tak(); return; } for (int i=0; i<n; ++i) { if (e[i].size() < 2) { if (a[i] == b[i]) tak(); else nie(); return; } } } bool same = true; for (int i=0; i<n; ++i) if (a[i] != b[i]) same = false; if (same) { tak(); return; } if (flipsA < 1) { nie(); return; } if (flipsB < n-1) tak(); else nie(); } int main() { int t; scanf("%d", &t); for (int s=0; s<t; ++s) solve(); return 0; } |
English