#include <iostream> #include <vector> using namespace std; vector <int> MT[1010]; vector <int> MN[1010]; vector <int> MTF[1010]; vector <int> MNF[1010]; vector <int> KO[1010]; int kolor[1010], xd, dzieci[1010], bycmozetata[1010]; bool mapkat[1010][1010], mapkan[1010][1010], permkolor[1010], bolek, mamyojca, niedasie; void DFS(int a) { kolor[a]=xd; KO[xd].push_back(a); for(int i=0; i<MT[a].size(); i++) { if(kolor[MT[a][i]]==0 && permkolor[MT[a][i]]==0) DFS(MT[a][i]); } for(int i=0; i<MTF[a].size(); i++) { if(kolor[MTF[a][i]]==0 && permkolor[MTF[a][i]]==0) DFS(MTF[a][i]); } } int main() { ios_base::sync_with_stdio(0); int pra, pre, a, b, ojciecdyrektor=-1, temp, temp2; char tn; cin>>pra>>pre; for(int i=0; i<pre; i++) { cin>>a>>b>>tn; if(tn=='T') { MT[a].push_back(b); MTF[b].push_back(a); mapkat[a][b]=1; } if(tn=='N') { MN[a].push_back(b); MNF[b].push_back(a); mapkan[a][b]=1; } } for(int i=1; i<=pra; i++) { if(MNF[i].size()==0 && MT[i].size()==0) { ojciecdyrektor=i; permkolor[i]=1; bolek=1; break; } } if(bolek==1) { for(int i=1; i<=pra; i++) { if(permkolor[i]==0) bycmozetata[i]=ojciecdyrektor; } xd=1; } if(bolek==0) cout<<"NIE"; else { while(xd>0) { xd=0; for(int i=1; i<=pra; i++) { if(kolor[i]==0 && permkolor[i]==0) { xd++; DFS(i); } } for(int i=1; i<=xd; i++) { ojciecdyrektor=-1; bolek=0; mamyojca=0; if(KO[i].size()==1) { dzieci[KO[i][0]]=bycmozetata[KO[i][0]]; permkolor[KO[i][0]]=1; KO[i].clear(); } else { ojciecdyrektor=-1; bolek=0; mamyojca=0; for(int j=0; j<KO[i].size(); j++) { bolek=0; temp=KO[i][j]; for(int k=0; k<MNF[temp].size(); k++) { temp2=MNF[temp][k]; if(permkolor[temp2]==0 && kolor[temp2]==kolor[temp]) { bolek=1; break; } } if(bolek==0) { for(int k=0; k<MT[temp].size(); k++) { temp2=MT[temp][k]; if(permkolor[temp2]==0 && kolor[temp2]==kolor[temp]) { bolek=1; break; } } } if(bolek==0) { mamyojca=1; ojciecdyrektor=temp; permkolor[temp]=1; break; } } if(mamyojca==1) { dzieci[ojciecdyrektor]=bycmozetata[ojciecdyrektor]; for(int j=0; j<KO[i].size(); j++) { if(KO[i][j]!=ojciecdyrektor) bycmozetata[KO[i][j]]=ojciecdyrektor; } KO[i].clear(); } else { niedasie=1; break; } } } if(niedasie==1) break; for(int i=0; i<=pra; i++) kolor[i]=0; } if(niedasie==1) cout<<"NIE"; else { for(int i=1; i<=pra; i++) cout<<dzieci[i]<<endl; } } }
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 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 | #include <iostream> #include <vector> using namespace std; vector <int> MT[1010]; vector <int> MN[1010]; vector <int> MTF[1010]; vector <int> MNF[1010]; vector <int> KO[1010]; int kolor[1010], xd, dzieci[1010], bycmozetata[1010]; bool mapkat[1010][1010], mapkan[1010][1010], permkolor[1010], bolek, mamyojca, niedasie; void DFS(int a) { kolor[a]=xd; KO[xd].push_back(a); for(int i=0; i<MT[a].size(); i++) { if(kolor[MT[a][i]]==0 && permkolor[MT[a][i]]==0) DFS(MT[a][i]); } for(int i=0; i<MTF[a].size(); i++) { if(kolor[MTF[a][i]]==0 && permkolor[MTF[a][i]]==0) DFS(MTF[a][i]); } } int main() { ios_base::sync_with_stdio(0); int pra, pre, a, b, ojciecdyrektor=-1, temp, temp2; char tn; cin>>pra>>pre; for(int i=0; i<pre; i++) { cin>>a>>b>>tn; if(tn=='T') { MT[a].push_back(b); MTF[b].push_back(a); mapkat[a][b]=1; } if(tn=='N') { MN[a].push_back(b); MNF[b].push_back(a); mapkan[a][b]=1; } } for(int i=1; i<=pra; i++) { if(MNF[i].size()==0 && MT[i].size()==0) { ojciecdyrektor=i; permkolor[i]=1; bolek=1; break; } } if(bolek==1) { for(int i=1; i<=pra; i++) { if(permkolor[i]==0) bycmozetata[i]=ojciecdyrektor; } xd=1; } if(bolek==0) cout<<"NIE"; else { while(xd>0) { xd=0; for(int i=1; i<=pra; i++) { if(kolor[i]==0 && permkolor[i]==0) { xd++; DFS(i); } } for(int i=1; i<=xd; i++) { ojciecdyrektor=-1; bolek=0; mamyojca=0; if(KO[i].size()==1) { dzieci[KO[i][0]]=bycmozetata[KO[i][0]]; permkolor[KO[i][0]]=1; KO[i].clear(); } else { ojciecdyrektor=-1; bolek=0; mamyojca=0; for(int j=0; j<KO[i].size(); j++) { bolek=0; temp=KO[i][j]; for(int k=0; k<MNF[temp].size(); k++) { temp2=MNF[temp][k]; if(permkolor[temp2]==0 && kolor[temp2]==kolor[temp]) { bolek=1; break; } } if(bolek==0) { for(int k=0; k<MT[temp].size(); k++) { temp2=MT[temp][k]; if(permkolor[temp2]==0 && kolor[temp2]==kolor[temp]) { bolek=1; break; } } } if(bolek==0) { mamyojca=1; ojciecdyrektor=temp; permkolor[temp]=1; break; } } if(mamyojca==1) { dzieci[ojciecdyrektor]=bycmozetata[ojciecdyrektor]; for(int j=0; j<KO[i].size(); j++) { if(KO[i][j]!=ojciecdyrektor) bycmozetata[KO[i][j]]=ojciecdyrektor; } KO[i].clear(); } else { niedasie=1; break; } } } if(niedasie==1) break; for(int i=0; i<=pra; i++) kolor[i]=0; } if(niedasie==1) cout<<"NIE"; else { for(int i=1; i<=pra; i++) cout<<dzieci[i]<<endl; } } } |