#include <bits/stdc++.h> #define int long long #define mp make_pair #define pb push_back #define ld long double #define pii pair<int,int> #define sz(x) (int)x.size() #define piii pair<pii,pii> #define precise cout<<fixed<<setprecision(10) #define st first #define nd second #define ins insert #define vi vector<int> #define BOOST ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0) using namespace std; const int MAX=1e5+5; char color1[MAX]; char color2[MAX]; int gle[MAX]; vi P[MAX]; int n; int sciezka(){ int root=-1; for (int i=1;i<=n;i++){ if (sz(P[i])==1)root=i; else if (sz(P[i])>2){ return -1; } } return root; } string func(int root,bool flaga){ int fa=-1; string ans=""; while (true){ if (!flaga){ char lit=color1[root]; if (ans.empty() || lit!=ans.back())ans.pb(lit); } else{ char lit=color2[root]; if (ans.empty() || lit!=ans.back())ans.pb(lit); } int prv=root; for (auto it:P[root])if (it!=fa)root=it; fa=prv; if (root==prv)break; } return ans; } vi v1,v2; void dfs(int u,int fa){ if (gle[u]%2==0)v1.pb(u); else v2.pb(u); for (auto it:P[u]){ if (it==fa)continue; gle[it]=gle[u]+1; dfs(it,u); } } void solve(){ v1.clear(),v2.clear(); cin>>n; for (int i=1;i<=n;i++)cin>>color1[i]; for (int i=1;i<=n;i++)cin>>color2[i],P[i].clear(); for (int i=1;i<=n-1;i++){ int a,b; cin>>a>>b; P[a].pb(b); P[b].pb(a); } if (sciezka()!=-1){ int root=sciezka(); string s1=func(root,0); string s2=func(root,1); int x=sz(s1),y=sz(s2); //cout<<"LEN "<<sz(s1)<<" "<<sz(s2)<<" "<<s1<<" "<<s2<<"\n"; if (s1[0]!=s2[0])x--; if (x>=y)cout<<"TAK\n"; else cout<<"NIE\n"; return; } bool allb1=true,allr1=true,allb2=true,allr2=true; for (int i=1;i<=n;i++){ if (color1[i]!='0')allb1=false; if (color1[i]!='1')allr1=false; if (color2[i]!='0')allb2=false; if (color2[i]!='1')allr2=false; } if ((allb1 && !allb2)){cout<<"NIE\n";return;} if ((allr1 && !allr2)){cout<<"NIE\n";return;} if ((allb1 && allb2)){cout<<"TAK\n";return;} if ((allr1 && allr2)){cout<<"TAK\n";return;} gle[1]=0; dfs(1,-1); bool d1a=true,d1b=true,d2a=true,d2b=true; // kolorowanie1 0-1 kolorowanie2 0-2 for (auto it:v1){ if (color1[it]!='0')d1a=false; if (color2[it]!='0')d2a=false; if (color1[it]!='1')d1b=false; if (color2[it]!='1')d2b=false; } for (auto it:v2){ if (color1[it]!='1')d1a=false; if (color2[it]!='1')d2a=false; if (color1[it]!='0')d1b=false; if (color2[it]!='0')d2b=false; } if (d1a && d2b){cout<<"NIE\n";return;} if (d1b && d2a){cout<<"NIE\n";return;} if ((!d1a && !d1b) && (d2a || d2b)){cout<<"NIE\n";return;} cout<<"TAK\n"; } int32_t main(){ BOOST; int t; cin>>t; for (int z=0;z<t;z++){ solve(); } 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 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 | #include <bits/stdc++.h> #define int long long #define mp make_pair #define pb push_back #define ld long double #define pii pair<int,int> #define sz(x) (int)x.size() #define piii pair<pii,pii> #define precise cout<<fixed<<setprecision(10) #define st first #define nd second #define ins insert #define vi vector<int> #define BOOST ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0) using namespace std; const int MAX=1e5+5; char color1[MAX]; char color2[MAX]; int gle[MAX]; vi P[MAX]; int n; int sciezka(){ int root=-1; for (int i=1;i<=n;i++){ if (sz(P[i])==1)root=i; else if (sz(P[i])>2){ return -1; } } return root; } string func(int root,bool flaga){ int fa=-1; string ans=""; while (true){ if (!flaga){ char lit=color1[root]; if (ans.empty() || lit!=ans.back())ans.pb(lit); } else{ char lit=color2[root]; if (ans.empty() || lit!=ans.back())ans.pb(lit); } int prv=root; for (auto it:P[root])if (it!=fa)root=it; fa=prv; if (root==prv)break; } return ans; } vi v1,v2; void dfs(int u,int fa){ if (gle[u]%2==0)v1.pb(u); else v2.pb(u); for (auto it:P[u]){ if (it==fa)continue; gle[it]=gle[u]+1; dfs(it,u); } } void solve(){ v1.clear(),v2.clear(); cin>>n; for (int i=1;i<=n;i++)cin>>color1[i]; for (int i=1;i<=n;i++)cin>>color2[i],P[i].clear(); for (int i=1;i<=n-1;i++){ int a,b; cin>>a>>b; P[a].pb(b); P[b].pb(a); } if (sciezka()!=-1){ int root=sciezka(); string s1=func(root,0); string s2=func(root,1); int x=sz(s1),y=sz(s2); //cout<<"LEN "<<sz(s1)<<" "<<sz(s2)<<" "<<s1<<" "<<s2<<"\n"; if (s1[0]!=s2[0])x--; if (x>=y)cout<<"TAK\n"; else cout<<"NIE\n"; return; } bool allb1=true,allr1=true,allb2=true,allr2=true; for (int i=1;i<=n;i++){ if (color1[i]!='0')allb1=false; if (color1[i]!='1')allr1=false; if (color2[i]!='0')allb2=false; if (color2[i]!='1')allr2=false; } if ((allb1 && !allb2)){cout<<"NIE\n";return;} if ((allr1 && !allr2)){cout<<"NIE\n";return;} if ((allb1 && allb2)){cout<<"TAK\n";return;} if ((allr1 && allr2)){cout<<"TAK\n";return;} gle[1]=0; dfs(1,-1); bool d1a=true,d1b=true,d2a=true,d2b=true; // kolorowanie1 0-1 kolorowanie2 0-2 for (auto it:v1){ if (color1[it]!='0')d1a=false; if (color2[it]!='0')d2a=false; if (color1[it]!='1')d1b=false; if (color2[it]!='1')d2b=false; } for (auto it:v2){ if (color1[it]!='1')d1a=false; if (color2[it]!='1')d2a=false; if (color1[it]!='0')d1b=false; if (color2[it]!='0')d2b=false; } if (d1a && d2b){cout<<"NIE\n";return;} if (d1b && d2a){cout<<"NIE\n";return;} if ((!d1a && !d1b) && (d2a || d2b)){cout<<"NIE\n";return;} cout<<"TAK\n"; } int32_t main(){ BOOST; int t; cin>>t; for (int z=0;z<t;z++){ solve(); } return 0; } |