// Autor: Mikołaj Janusz #include <bits/stdc++.h> typedef long long ll; using namespace std; bool bfs(ll x, vector<vector<ll>>& graph, vector<bool>& colors, vector<bool>& targets, set<ll>& visited) { if (colors[x] == targets[x]) { return true; } queue<ll> q; q.push(x); visited.insert(x); while (!q.empty()) { ll cur = q.front(); q.pop(); for (ll i = 0; i != graph[cur].size(); i++) { if (graph[cur][i] && (visited.find(i) == visited.end())) { if (colors[i] == targets[x]) { if (bfs(i, graph, colors, targets, visited)) { return true; } } q.push(i); visited.insert(i); } } } return false; } int main() { ios::sync_with_stdio(false); ll tests; cin >> tests; for (ll t = 0; t < tests; t++) { ll n; cin >> n; string s_col; string s_tar; string result = "TAK"; vector<vector<ll>> graph(n + 3); vector<bool> colors(n+1); vector<bool> targets(n+1); cin >> s_col >> s_tar; for (ll i = 1; i <= n; i++) { colors[i] = (s_col[i-1] == '1'); targets[i] = (s_tar[i-1] == '1'); } for (ll i = 0; i < n - 1; i++) { ll x, y; cin >> x >> y; graph[x].push_back(y); graph[y].push_back(x); } for (ll i = 1; i <= n; i++) { if (colors[i] != targets[i]) { set<ll> visited; if (!bfs(i, graph, colors, targets, visited)) { result = "NIE"; break; } } } cout << result << endl; } 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 | // Autor: Mikołaj Janusz #include <bits/stdc++.h> typedef long long ll; using namespace std; bool bfs(ll x, vector<vector<ll>>& graph, vector<bool>& colors, vector<bool>& targets, set<ll>& visited) { if (colors[x] == targets[x]) { return true; } queue<ll> q; q.push(x); visited.insert(x); while (!q.empty()) { ll cur = q.front(); q.pop(); for (ll i = 0; i != graph[cur].size(); i++) { if (graph[cur][i] && (visited.find(i) == visited.end())) { if (colors[i] == targets[x]) { if (bfs(i, graph, colors, targets, visited)) { return true; } } q.push(i); visited.insert(i); } } } return false; } int main() { ios::sync_with_stdio(false); ll tests; cin >> tests; for (ll t = 0; t < tests; t++) { ll n; cin >> n; string s_col; string s_tar; string result = "TAK"; vector<vector<ll>> graph(n + 3); vector<bool> colors(n+1); vector<bool> targets(n+1); cin >> s_col >> s_tar; for (ll i = 1; i <= n; i++) { colors[i] = (s_col[i-1] == '1'); targets[i] = (s_tar[i-1] == '1'); } for (ll i = 0; i < n - 1; i++) { ll x, y; cin >> x >> y; graph[x].push_back(y); graph[y].push_back(x); } for (ll i = 1; i <= n; i++) { if (colors[i] != targets[i]) { set<ll> visited; if (!bfs(i, graph, colors, targets, visited)) { result = "NIE"; break; } } } cout << result << endl; } return 0; } |