#include <iostream> #include <fstream> #include <algorithm> #include <vector> using namespace std; struct wier{ vector<wier*> sas; bool poczatkowy; bool docelowy; }; struct Graf{ int n; wier* ziomki; void wczytaj(){ scanf("%d", &n); char cpom=0; ziomki = new wier[n]; for(int i=0; i<n; i++){ cpom=0; while(cpom!='0' && cpom!='1') scanf("%c", &cpom); ziomki[i].poczatkowy=(cpom=='1'); } for(int i=0; i<n; i++){ cpom=0; while(cpom!='0' && cpom!='1') scanf("%c", &cpom); ziomki[i].docelowy=(cpom=='1'); } int a, b; for(int i=1; i<n; i++){ scanf("%d %d", &a, &b); a--; b--; ziomki[a].sas.push_back(ziomki+b); ziomki[b].sas.push_back(ziomki+a); } } bool czy_pasiasty(){ for(int i=0; i<n; i++) for(int j=0; j<ziomki[i].sas.size(); j++) if(ziomki[i].docelowy==ziomki[i].sas[j]->docelowy) return 0; return 1; } char przypadek_prostej(){ ///-1: to nie prosta 0: nie da sie 1: da sie wier ** prosta = new wier*[n]; for(int i=0; i<n; i++) if(ziomki[i].sas.size()>2) return -1; else if(ziomki[i].sas.size()==1) prosta[0]=ziomki+i; for(int i=1; i<n; i++){ prosta[i] = prosta[i-1]->sas[0]; if(i>=2 && prosta[i]==prosta[i-2]) prosta[i] = prosta[i-1]->sas[1]; } int zmianyp=0; int zmianyk=0; for(int i=1; i<n; i++){ zmianyp+=(prosta[i]->poczatkowy!=prosta[i-1]->poczatkowy); zmianyk+=(prosta[i]->docelowy!=prosta[i-1]->docelowy); } char zwrot=0; if(zmianyp>zmianyk || (zmianyp==zmianyk && prosta[0]->poczatkowy==prosta[0]->docelowy)) zwrot= 1; // delete[] prosta; return zwrot; } bool unikolor(){ ///0: tak, zero 1: tak, jeden 2: nie (poczatkowy+4*docelowy); bool czyuni=1; for(int i=0; i<n; i++) if(ziomki[i].poczatkowy!=ziomki[0].poczatkowy) czyuni=0; return czyuni; } bool czytriw(){ for(int i=0; i<n; i++) if(ziomki[i].docelowy!=ziomki[i].poczatkowy) return 0; return 1; } bool rozw(){ if(czytriw()) return 1; if(unikolor()) return 0; char cp=przypadek_prostej(); if(cp!=-1) return cp; if(czy_pasiasty()) return 0; return 1; } }; int main() { int q; scanf("%d", &q); for(int i=0; i<q; i++){ Graf graf; graf.wczytaj(); if(graf.rozw()) printf("TAK\n"); else printf("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 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 | #include <iostream> #include <fstream> #include <algorithm> #include <vector> using namespace std; struct wier{ vector<wier*> sas; bool poczatkowy; bool docelowy; }; struct Graf{ int n; wier* ziomki; void wczytaj(){ scanf("%d", &n); char cpom=0; ziomki = new wier[n]; for(int i=0; i<n; i++){ cpom=0; while(cpom!='0' && cpom!='1') scanf("%c", &cpom); ziomki[i].poczatkowy=(cpom=='1'); } for(int i=0; i<n; i++){ cpom=0; while(cpom!='0' && cpom!='1') scanf("%c", &cpom); ziomki[i].docelowy=(cpom=='1'); } int a, b; for(int i=1; i<n; i++){ scanf("%d %d", &a, &b); a--; b--; ziomki[a].sas.push_back(ziomki+b); ziomki[b].sas.push_back(ziomki+a); } } bool czy_pasiasty(){ for(int i=0; i<n; i++) for(int j=0; j<ziomki[i].sas.size(); j++) if(ziomki[i].docelowy==ziomki[i].sas[j]->docelowy) return 0; return 1; } char przypadek_prostej(){ ///-1: to nie prosta 0: nie da sie 1: da sie wier ** prosta = new wier*[n]; for(int i=0; i<n; i++) if(ziomki[i].sas.size()>2) return -1; else if(ziomki[i].sas.size()==1) prosta[0]=ziomki+i; for(int i=1; i<n; i++){ prosta[i] = prosta[i-1]->sas[0]; if(i>=2 && prosta[i]==prosta[i-2]) prosta[i] = prosta[i-1]->sas[1]; } int zmianyp=0; int zmianyk=0; for(int i=1; i<n; i++){ zmianyp+=(prosta[i]->poczatkowy!=prosta[i-1]->poczatkowy); zmianyk+=(prosta[i]->docelowy!=prosta[i-1]->docelowy); } char zwrot=0; if(zmianyp>zmianyk || (zmianyp==zmianyk && prosta[0]->poczatkowy==prosta[0]->docelowy)) zwrot= 1; // delete[] prosta; return zwrot; } bool unikolor(){ ///0: tak, zero 1: tak, jeden 2: nie (poczatkowy+4*docelowy); bool czyuni=1; for(int i=0; i<n; i++) if(ziomki[i].poczatkowy!=ziomki[0].poczatkowy) czyuni=0; return czyuni; } bool czytriw(){ for(int i=0; i<n; i++) if(ziomki[i].docelowy!=ziomki[i].poczatkowy) return 0; return 1; } bool rozw(){ if(czytriw()) return 1; if(unikolor()) return 0; char cp=przypadek_prostej(); if(cp!=-1) return cp; if(czy_pasiasty()) return 0; return 1; } }; int main() { int q; scanf("%d", &q); for(int i=0; i<q; i++){ Graf graf; graf.wczytaj(); if(graf.rozw()) printf("TAK\n"); else printf("NIE\n"); } return 0; } |