Pagini recente » Cod sursa (job #1682693) | Cod sursa (job #322425) | Cod sursa (job #3189096) | Cod sursa (job #873721) | Cod sursa (job #2007280)
/*
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],R[100001];
vector<int> V[100001];
void dfs(int nod, int p)
{
int i;
nr++;
for(i=0; i<V[nod].size(); i++)
{
if(V[nod][i]!=p)
{
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],nod);
}
}
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);
V[y].push_back(x);
}
S[1]=0;
nr=1;
dfs(1,0);
for(i=1; i<=n; i++)
fo<<R[i]<<" ";
fo<<"\n";
fi.close();
fo.close();
return 0;
}