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