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