#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; } |
English