Pagini recente » Cod sursa (job #2915552) | Cod sursa (job #1856973) | Cod sursa (job #2589906) | Cod sursa (job #2760854) | Cod sursa (job #2357493)
#include <fstream>
#include <vector>
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
int n,viz[100003],st[100003],k[100003],sol[100003],dim;
vector<int>g[100003];
void dfs(int nod)
{
st[++dim]=nod;
int i=nod;
if(k[i])
{
sol[i]=sol[st[dim-k[i]]]+1;
}
for(vector<int> :: iterator it=g[nod].begin();it!=g[nod].end();it++)
dfs(*it);
dim--;
}
int main()
{
fin>>n;
for(int i=1;i<=n;i++)
fin>>k[i];
for(int i=1;i<n;i++)
{
int x,y;
fin>>x>>y;
g[x].push_back(y);
viz[y]++;
}
for(int i=1;i<=n;i++)
if(viz[i]==0)
{
dfs(i);
break;
}
for(int i=1;i<=n;i++)
fout<<sol[i]<<" ";
}