#define _USE_MATH_DEFINES
#include <bits/stdc++.h>
#ifdef LOC
#include "debuglib.h"
#else
#define deb(...)
#define DBP(...)
#endif
using namespace std;
using ll = long long;
using Vi = vector<int>;
using Pii = pair<int, int>;
#define pb push_back
#define mp make_pair
#define x first
#define y second
#define rep(i, b, e) for (int i = (b); i < (e); i++)
#define each(a, x) for (auto& a : (x))
#define all(x) (x).begin(), (x).end()
#define sz(x) int((x).size())
bool run() {
int n; cin >> n;
string s, t;
cin >> s >> t;
vector<Vi> G(n);
rep(i, 1, n) {
int u, v; cin >> u >> v;
G[u-1].pb(v-1);
G[v-1].pb(u-1);
}
if (s == t) {
return true;
}
int k = int(count(all(s), '0'));
if (k == 0 || k == n) {
return false;
}
rep(i, 0, n) {
if (sz(G[i]) >= 3) {
rep(v, 0, n) each(e, G[v]) if (t[v] == t[e]) return true;
return false;
}
}
Vi path;
rep(i, 0, n) if (sz(G[i]) == 1) {
path.pb(i);
path.pb(G[i][0]);
break;
}
while (sz(G[path.back()]) != 1) {
auto& vec = G[path.back()];
path.pb(vec[vec[0] == path[sz(path)-2]]);
}
int segs1 = 1, segs2 = 1;
rep(i, 1, sz(path)) {
int u = path[i-1], v = path[i];
segs1 += s[u] != s[v];
segs2 += t[u] != t[v];
}
segs1 -= (s[path[0]] != t[path[0]]);
segs1 -= (s[path.back()] != t[path.back()]);
return segs1 >= segs2;
}
int main() {
cin.sync_with_stdio(0); cin.tie(0);
cout << fixed << setprecision(12);
int t; cin >> t;
while (t--) cout << (run() ? "TAK\n" : "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 | #define _USE_MATH_DEFINES #include <bits/stdc++.h> #ifdef LOC #include "debuglib.h" #else #define deb(...) #define DBP(...) #endif using namespace std; using ll = long long; using Vi = vector<int>; using Pii = pair<int, int>; #define pb push_back #define mp make_pair #define x first #define y second #define rep(i, b, e) for (int i = (b); i < (e); i++) #define each(a, x) for (auto& a : (x)) #define all(x) (x).begin(), (x).end() #define sz(x) int((x).size()) bool run() { int n; cin >> n; string s, t; cin >> s >> t; vector<Vi> G(n); rep(i, 1, n) { int u, v; cin >> u >> v; G[u-1].pb(v-1); G[v-1].pb(u-1); } if (s == t) { return true; } int k = int(count(all(s), '0')); if (k == 0 || k == n) { return false; } rep(i, 0, n) { if (sz(G[i]) >= 3) { rep(v, 0, n) each(e, G[v]) if (t[v] == t[e]) return true; return false; } } Vi path; rep(i, 0, n) if (sz(G[i]) == 1) { path.pb(i); path.pb(G[i][0]); break; } while (sz(G[path.back()]) != 1) { auto& vec = G[path.back()]; path.pb(vec[vec[0] == path[sz(path)-2]]); } int segs1 = 1, segs2 = 1; rep(i, 1, sz(path)) { int u = path[i-1], v = path[i]; segs1 += s[u] != s[v]; segs2 += t[u] != t[v]; } segs1 -= (s[path[0]] != t[path[0]]); segs1 -= (s[path.back()] != t[path.back()]); return segs1 >= segs2; } int main() { cin.sync_with_stdio(0); cin.tie(0); cout << fixed << setprecision(12); int t; cin >> t; while (t--) cout << (run() ? "TAK\n" : "NIE\n"); return 0; } |
English