#define make_pair mp
#define emplace_back pb
#include <bits/stdc++.h>
using namespace std;
mt19937 mt_rand(time(0));
const int N = 1e5 + 5;
int n, t, perm[N];
vector<int> v[N];
bool init[N], dest[N];
char s[N];
int main() {
scanf("%d", &t);
while(t--) {
for(int i=1;i<=n;i++) v[i].clear();
scanf("%d", &n);
scanf("%s", s);
for(int i=0;i<n;i++) init[i+1] = s[i] - '0';
scanf("%s", s);
for(int i=0;i<n;i++) dest[i+1] = s[i] - '0';
for(int i=1;i<n;i++) {
int a, b;
scanf("%d%d", &a, &b);
v[a].pb(b);
v[b].pb(a);
}
int cnt = 0;
for(int i=1;i<=n;i++) cnt += init[i];
if(cnt == 0 || cnt == n) {
int cnt2 = 0;
for(int i=1;i<=n;i++) cnt2 += dest[i];
if(cnt == cnt2) printf("TAK\n");
else printf("NIE\n");
continue;
}
int maks = 0;
for(int i=1;i<=n;i++) maks = max(maks, (int)v[i].size());
if(maks > 2) {
bool ok = true;
for(int i=1;i<=n;i++)
for(auto x : v[i]) if(dest[i] == dest[x]) ok = false;
if(!ok) {
printf("TAK\n");
continue;
}
ok = true;
for(int i=1;i<=n;i++) if(init[i] != dest[i]) {
printf("NIE\n");
ok = false;
break;
}
if(ok) printf("TAK\n");
continue;
}
int root = 1;
while(v[root].size() == 2) root++;
perm[1] = root;
for(int i=2;i<=n;i++) {
int prev = perm[i-1];
int prev2 = perm[i-2];
for(auto x : v[prev]) if(x != prev2) {
perm[i] = x;
break;
}
}
cnt = 0;
int cnt2 = 0;
perm[n+1] = 0;
for(int i=1;i<=n+1;i++) if(init[perm[i]] == 0 && init[perm[i-1]] == 1) cnt++;
for(int i=1;i<=n+1;i++) if(dest[perm[i]] == 0 && dest[perm[i-1]] == 1) cnt2++;
if(init[perm[1]] == 1 && dest[perm[1]] == 0) cnt--;
if(init[perm[n]] == 1 && dest[perm[n]] == 0) cnt--;
if(cnt >= cnt2) printf("TAK\n");
else printf("NIE\n");
}
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 | #define make_pair mp #define emplace_back pb #include <bits/stdc++.h> using namespace std; mt19937 mt_rand(time(0)); const int N = 1e5 + 5; int n, t, perm[N]; vector<int> v[N]; bool init[N], dest[N]; char s[N]; int main() { scanf("%d", &t); while(t--) { for(int i=1;i<=n;i++) v[i].clear(); scanf("%d", &n); scanf("%s", s); for(int i=0;i<n;i++) init[i+1] = s[i] - '0'; scanf("%s", s); for(int i=0;i<n;i++) dest[i+1] = s[i] - '0'; for(int i=1;i<n;i++) { int a, b; scanf("%d%d", &a, &b); v[a].pb(b); v[b].pb(a); } int cnt = 0; for(int i=1;i<=n;i++) cnt += init[i]; if(cnt == 0 || cnt == n) { int cnt2 = 0; for(int i=1;i<=n;i++) cnt2 += dest[i]; if(cnt == cnt2) printf("TAK\n"); else printf("NIE\n"); continue; } int maks = 0; for(int i=1;i<=n;i++) maks = max(maks, (int)v[i].size()); if(maks > 2) { bool ok = true; for(int i=1;i<=n;i++) for(auto x : v[i]) if(dest[i] == dest[x]) ok = false; if(!ok) { printf("TAK\n"); continue; } ok = true; for(int i=1;i<=n;i++) if(init[i] != dest[i]) { printf("NIE\n"); ok = false; break; } if(ok) printf("TAK\n"); continue; } int root = 1; while(v[root].size() == 2) root++; perm[1] = root; for(int i=2;i<=n;i++) { int prev = perm[i-1]; int prev2 = perm[i-2]; for(auto x : v[prev]) if(x != prev2) { perm[i] = x; break; } } cnt = 0; int cnt2 = 0; perm[n+1] = 0; for(int i=1;i<=n+1;i++) if(init[perm[i]] == 0 && init[perm[i-1]] == 1) cnt++; for(int i=1;i<=n+1;i++) if(dest[perm[i]] == 0 && dest[perm[i-1]] == 1) cnt2++; if(init[perm[1]] == 1 && dest[perm[1]] == 0) cnt--; if(init[perm[n]] == 1 && dest[perm[n]] == 0) cnt--; if(cnt >= cnt2) printf("TAK\n"); else printf("NIE\n"); } return 0; } |
English