Pagini recente » Cod sursa (job #2439780) | Cod sursa (job #2696874) | Cod sursa (job #1192753) | Cod sursa (job #2683820) | Cod sursa (job #2007290)
/*
In timpul DFS-ului folosesc o stiva a.i. stramosul al k-lea al lui nr este S[nr-k].
*/
#include<fstream>
#include<vector>
using namespace std;
ifstream fi("cerere.in");
ofstream fo("cerere.out");
int n,i,nr,x,y,K[100001],S[100001],P[100001],R[100001];
vector<int> V[100001];
void dfs(int nod)
{
int i;
nr++;
for(i=0; i<V[nod].size(); i++)
{
if(K[V[nod][i]]!=0)
S[nr]=S[nr-K[V[nod][i]]]+1;
else
S[nr]=0;
R[V[nod][i]]=S[nr];
dfs(V[nod][i]);
}
S[nr]=0;
nr--;
}
int main()
{
fi>>n;
for(i=1; i<=n; i++)
{
fi>>K[i];
}
for(i=1; i<=n; i++)
{
fi>>x>>y;
V[x].push_back(y);
P[y]=x;
}
for(i=1; i<=n; i++)
{
if(P[i]==0)
{
S[i]=0;
nr=1;
dfs(i);
}
}
for(i=1; i<=n; i++)
fo<<R[i]<<" ";
fo<<"\n";
fi.close();
fo.close();
return 0;
}