/*
* Copyright (C) 2021 Paweł Widera
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details:
* http://www.gnu.org/licenses/gpl.html
*/
#include <iostream>
#include <algorithm>
#include <vector>
#include <string>
using namespace std;
#ifdef DEBUG
template<typename... ArgTypes>
inline void debug_print(ArgTypes... args) {
using expand = int[];
(void)expand{0, ((cerr << args << " "), void(), 0)... };
cerr << endl;
}
#define DBG(...) debug_print(__VA_ARGS__);
#define DBGx(x) cerr << #x << " " << x << endl;
#define DBGa(array) cerr << #array << " "; for (auto a: array) { cerr << a << " "; } cerr << endl;
#else
#define DBG(...)
#define DBGx(x)
#define DBGa(array)
#endif
string solve(vector<bool>& source, vector<int>& different, vector<vector<int>>& graph) {
if (different.size() == 0) {
return "NIE";
}
return "TAK";
}
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
int t;
cin >> t;
int n, a, b;
char colour;
vector<bool> source; // true if red, false if black
vector<int> different;
while (t-- > 0) {
cin >> n;
source.clear();
for (int i = 0; i < n; ++i) {
cin >> colour;
source.push_back(colour == '0');
}
// nodes to flip
different.clear();
for (int i = 0; i < n; ++i) {
cin >> colour;
if (source[i] && colour == '1') {
different.push_back(i);
}
}
// tree edges
vector<vector<int>> graph(n);
for (int i = 0; i < n - 1; ++i) {
cin >> a >> b;
graph[a - 1].push_back(b - 1);
graph[b - 1].push_back(a - 1);
}
cout << solve(source, different, graph) << 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 78 79 80 81 82 83 84 85 86 87 88 | /* * Copyright (C) 2021 Paweł Widera * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 3 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details: * http://www.gnu.org/licenses/gpl.html */ #include <iostream> #include <algorithm> #include <vector> #include <string> using namespace std; #ifdef DEBUG template<typename... ArgTypes> inline void debug_print(ArgTypes... args) { using expand = int[]; (void)expand{0, ((cerr << args << " "), void(), 0)... }; cerr << endl; } #define DBG(...) debug_print(__VA_ARGS__); #define DBGx(x) cerr << #x << " " << x << endl; #define DBGa(array) cerr << #array << " "; for (auto a: array) { cerr << a << " "; } cerr << endl; #else #define DBG(...) #define DBGx(x) #define DBGa(array) #endif string solve(vector<bool>& source, vector<int>& different, vector<vector<int>>& graph) { if (different.size() == 0) { return "NIE"; } return "TAK"; } int main() { ios::sync_with_stdio(false); cin.tie(nullptr); int t; cin >> t; int n, a, b; char colour; vector<bool> source; // true if red, false if black vector<int> different; while (t-- > 0) { cin >> n; source.clear(); for (int i = 0; i < n; ++i) { cin >> colour; source.push_back(colour == '0'); } // nodes to flip different.clear(); for (int i = 0; i < n; ++i) { cin >> colour; if (source[i] && colour == '1') { different.push_back(i); } } // tree edges vector<vector<int>> graph(n); for (int i = 0; i < n - 1; ++i) { cin >> a >> b; graph[a - 1].push_back(b - 1); graph[b - 1].push_back(a - 1); } cout << solve(source, different, graph) << endl; } } |
English