#include <stdio.h> #include <string> #include <vector> #include <set> #include <string.h> using std::string; using std::vector; using std::pair; using std::make_pair; using std::set; const int max_size = 1e5; int n; char start_coding[max_size+1]; char end_coding[max_size+1]; vector<pair<int, int>> trees_edges; bool one_color() { for (int i = 1; i < n; i++) if (start_coding[i] != start_coding[i-1]) return false; return true; } bool linia_wynik; bool linia() { int start_segments = 0; int end_segments = 0; static int neighbours[max_size]; for (int i = 0; i < n; i++) neighbours[i] = 0; for (auto v : trees_edges) { neighbours[v.first]++; neighbours[v.second]++; if (start_coding[v.first] != start_coding[v.second]) start_segments++; if (end_coding[v.first] != end_coding[v.second]) end_segments++; } for (int i = 0; i < n; i++) { if (neighbours[i] > 2) return false; if (neighbours[i] == 1) if (start_coding[i] != end_coding[i]) start_segments--; } linia_wynik = (start_segments >= end_segments); return true; } bool bipartite() { for (auto p : trees_edges) if (end_coding[p.first] == end_coding[p.second]) return false; return true; } int main() { int T; scanf("%d", &T); while(T--) { trees_edges.clear(); scanf("%d%s%s", &n, start_coding, end_coding); for (int i = 1; i < n; i++) { int a, b; scanf("%d%d", &a, &b); trees_edges.push_back(make_pair(a-1, b-1)); } if (strcmp(start_coding, end_coding) == 0) printf("TAK\n"); else if (one_color()) printf("NIE\n"); else if (linia()) printf(linia_wynik ? "TAK\n" : "NIE\n"); else if (bipartite()) printf("NIE\n"); else printf("TAK\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 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 | #include <stdio.h> #include <string> #include <vector> #include <set> #include <string.h> using std::string; using std::vector; using std::pair; using std::make_pair; using std::set; const int max_size = 1e5; int n; char start_coding[max_size+1]; char end_coding[max_size+1]; vector<pair<int, int>> trees_edges; bool one_color() { for (int i = 1; i < n; i++) if (start_coding[i] != start_coding[i-1]) return false; return true; } bool linia_wynik; bool linia() { int start_segments = 0; int end_segments = 0; static int neighbours[max_size]; for (int i = 0; i < n; i++) neighbours[i] = 0; for (auto v : trees_edges) { neighbours[v.first]++; neighbours[v.second]++; if (start_coding[v.first] != start_coding[v.second]) start_segments++; if (end_coding[v.first] != end_coding[v.second]) end_segments++; } for (int i = 0; i < n; i++) { if (neighbours[i] > 2) return false; if (neighbours[i] == 1) if (start_coding[i] != end_coding[i]) start_segments--; } linia_wynik = (start_segments >= end_segments); return true; } bool bipartite() { for (auto p : trees_edges) if (end_coding[p.first] == end_coding[p.second]) return false; return true; } int main() { int T; scanf("%d", &T); while(T--) { trees_edges.clear(); scanf("%d%s%s", &n, start_coding, end_coding); for (int i = 1; i < n; i++) { int a, b; scanf("%d%d", &a, &b); trees_edges.push_back(make_pair(a-1, b-1)); } if (strcmp(start_coding, end_coding) == 0) printf("TAK\n"); else if (one_color()) printf("NIE\n"); else if (linia()) printf(linia_wynik ? "TAK\n" : "NIE\n"); else if (bipartite()) printf("NIE\n"); else printf("TAK\n"); } return 0; } |