Pagini recente » Cod sursa (job #2225782) | Cod sursa (job #285111) | Cod sursa (job #477936) | Cod sursa (job #569470) | Cod sursa (job #2955813)
#include <fstream>
#include <vector>
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
int k[100001], t[100001], s[100001], sol[100001];
vector <int> l[100001];
void dfs(int nod, int niv)
{
s[niv]=nod;
if(k[nod]==0)
{
sol[nod]=0;
}
else
{
sol[nod]=sol[s[niv-k[nod]]]+1;
}
for(int i=0;i<l[nod].size();i++)
{
dfs(l[nod][i], niv+1);
}
}
int main()
{
int n, i, x, y, xs;
fin>>n;
for(i=1;i<=n;i++)
{
fin>>k[i];
}
for(i=1;i<n;i++)
{
fin>>x>>y;
l[x].push_back(y);
t[y]=x;
}
for(i=1;i<=n;i++)
{
if(t[i]==0)
{
xs=i;
break;
}
}
dfs(xs, 1);
for(i=1;i<=n;i++)
{
fout<<sol[i]<<" ";
}
}