#include <iostream>
#include <vector>
#include <queue>
#include <algorithm>
#include <set>
int n,a,b,t;
char x;
int main(){
std::ios_base::sync_with_stdio(0);
std::cin>>t;
for(int tt=0;tt<t;tt++){
std::vector<bool> s[2];
std::vector<int> st;
std::cin>>n;
st.resize(n+1,0);
int r[2]={0,0};
for(int j=0;j<2;j++)
for(int i=0;i<n;i++){
std::cin>>x;
s[j].emplace_back(x=='0');
}
int max_st=0;
int lisc;
for(int i=0;i<n-1;i++){
std::cin>>a>>b;
a--;b--;
st[a]++;
st[b]++;
max_st=std::max(max_st,std::max(st[a],st[b]));
for(int j=0;j<2;j++)
r[j]+=(s[j][a]==s[j][b]);
}
for(int i=0;i<n;i++)
if(st[i]==1)
lisc=i;
if(max_st==2){
if(r[0]>r[1]||(r[0]==r[1]&&s[0][lisc]!=s[1][lisc]))
std::cout<<"NIE"<<std::endl;
else
std::cout<<"TAK"<<std::endl;
}else{
if((r[1]>0 && (r[0]!=n-1 || (r[0]==r[1]&&s[0][0]==s[1][0])))||
(r[0]==0 && s[0][0]==s[1][0]))
std::cout<<"TAK"<<std::endl;
else
std::cout<<"NIE"<<std::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 | #include <iostream> #include <vector> #include <queue> #include <algorithm> #include <set> int n,a,b,t; char x; int main(){ std::ios_base::sync_with_stdio(0); std::cin>>t; for(int tt=0;tt<t;tt++){ std::vector<bool> s[2]; std::vector<int> st; std::cin>>n; st.resize(n+1,0); int r[2]={0,0}; for(int j=0;j<2;j++) for(int i=0;i<n;i++){ std::cin>>x; s[j].emplace_back(x=='0'); } int max_st=0; int lisc; for(int i=0;i<n-1;i++){ std::cin>>a>>b; a--;b--; st[a]++; st[b]++; max_st=std::max(max_st,std::max(st[a],st[b])); for(int j=0;j<2;j++) r[j]+=(s[j][a]==s[j][b]); } for(int i=0;i<n;i++) if(st[i]==1) lisc=i; if(max_st==2){ if(r[0]>r[1]||(r[0]==r[1]&&s[0][lisc]!=s[1][lisc])) std::cout<<"NIE"<<std::endl; else std::cout<<"TAK"<<std::endl; }else{ if((r[1]>0 && (r[0]!=n-1 || (r[0]==r[1]&&s[0][0]==s[1][0])))|| (r[0]==0 && s[0][0]==s[1][0])) std::cout<<"TAK"<<std::endl; else std::cout<<"NIE"<<std::endl; } } return 0; } |
English