#include <bits/stdc++.h> using namespace std; #define st first #define nd second #define iamspeed ios_base :: sync_with_stdio(0); cin.tie(0); cout.tie(0) int z, maxw, n, w[100005], ob[100005], doc[100005], vis[100005], a, b; string s1, s2; vector<int> G[100005], A, B; void DFS(int w, int ind) { vis[w] = 1; ob[ind] = (int)(s1[w - 1] - '0'); doc[ind] = (int)(s2[w - 1] - '0'); for(int i = 0 ; i < G[w].size() ; i++) { if(!vis[G[w][i]]) DFS(G[w][i], ind + 1); } } int main() { iamspeed; cin >> z; for(int k = 1 ; k <= z ; k++) { A.clear() ; B.clear(); maxw = 0; cin >> n; for(int i = 1 ; i <= n ; i++) { w[i] = vis[i] = 0; G[i].clear(); } cin >> s1 >> s2; for(int i = 1 ; i < n ; i++) { cin >> a >> b; G[a].push_back(b); G[b].push_back(a); w[a]++; w[b]++; maxw = max(maxw, max(w[a], w[b])); } if(maxw < 3) { for(int i = 1 ; i <= n ; i++) { if(w[i] == 1) { DFS(i, 1); break; } } ob[0] = doc[0] = -1; for(int i = 1 ; i <= n ; i++) { if(ob[i] != ob[i - 1]) A.push_back(ob[i]); if(doc[i] != doc[i - 1]) B.push_back(doc[i]); } if(A[0] != B[0]) B.push_back(1); if(A.size() >= B.size()) { cout << "TAK" << endl; } else { cout << "NIE" << endl; } } else { cout << "TAK" << endl; } } }
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 | #include <bits/stdc++.h> using namespace std; #define st first #define nd second #define iamspeed ios_base :: sync_with_stdio(0); cin.tie(0); cout.tie(0) int z, maxw, n, w[100005], ob[100005], doc[100005], vis[100005], a, b; string s1, s2; vector<int> G[100005], A, B; void DFS(int w, int ind) { vis[w] = 1; ob[ind] = (int)(s1[w - 1] - '0'); doc[ind] = (int)(s2[w - 1] - '0'); for(int i = 0 ; i < G[w].size() ; i++) { if(!vis[G[w][i]]) DFS(G[w][i], ind + 1); } } int main() { iamspeed; cin >> z; for(int k = 1 ; k <= z ; k++) { A.clear() ; B.clear(); maxw = 0; cin >> n; for(int i = 1 ; i <= n ; i++) { w[i] = vis[i] = 0; G[i].clear(); } cin >> s1 >> s2; for(int i = 1 ; i < n ; i++) { cin >> a >> b; G[a].push_back(b); G[b].push_back(a); w[a]++; w[b]++; maxw = max(maxw, max(w[a], w[b])); } if(maxw < 3) { for(int i = 1 ; i <= n ; i++) { if(w[i] == 1) { DFS(i, 1); break; } } ob[0] = doc[0] = -1; for(int i = 1 ; i <= n ; i++) { if(ob[i] != ob[i - 1]) A.push_back(ob[i]); if(doc[i] != doc[i - 1]) B.push_back(doc[i]); } if(A[0] != B[0]) B.push_back(1); if(A.size() >= B.size()) { cout << "TAK" << endl; } else { cout << "NIE" << endl; } } else { cout << "TAK" << endl; } } } |