Pagini recente » Cod sursa (job #362355) | Cod sursa (job #1144704) | Cod sursa (job #1235100) | Cod sursa (job #1183082) | Cod sursa (job #1826530)
#include<bits/stdc++.h>
#define maxN 100005
#define dim 1000005
using namespace std;
int seen[maxN],t[maxN],k[maxN],sol[maxN],n,x,y,viz[maxN],radacina;
vector<int> v[maxN];
char buff[dim+5];
int poz=0;
void read(int &nr)
{
nr=0;
while(buff[poz]<'0' || buff[poz]>'9')
{
poz++;
if(poz==dim)
{
poz=0;
fread(buff,1,dim,stdin);
}
}
while(buff[poz]>='0' && buff[poz]<='9')
{
nr=nr*10+buff[poz]-'0';
poz++;
if(poz==dim)
{
poz=0;
fread(buff,1,dim,stdin);
}
}
}
void dfs(int nod,int niv)
{
seen[nod]=1;
t[niv]=nod;
if(k[nod]) sol[nod]=1+sol[t[niv-k[nod]]];
else sol[nod]=0;
for(vector<int>::iterator it=v[nod].begin();it!=v[nod].end();it++)
{
if(!seen[*it])
dfs(*it,niv+1);
}
}
int main()
{
freopen("cerere.in","r",stdin);
freopen("cerere.out","w",stdout);
fread(buff,1,dim,stdin);
read(n);
for(int i=1;i<=n;i++)
read(k[i]);
for(int i=1;i<n;i++)
{
read(x);
read(y);
v[x].push_back(y);
v[y].push_back(x);
viz[y]=1;
}
for(int i=1;i<=n;i++)
{
if(!viz[i])
{
radacina=i;
break;
}
}
dfs(radacina,1);
for(int i=1;i<=n;i++)
{
printf("%d ",sol[i]);
}
return 0;
}