Pagini recente » Cod sursa (job #1979625) | Cod sursa (job #2708841) | Cod sursa (job #1196563) | Cod sursa (job #2811207) | Cod sursa (job #2780751)
#include <bits/stdc++.h>
using namespace std;
ifstream f("cerere.in");
ofstream g("cerere.out");
const int N=1e5+10;
vector <int> a[N];
int tata[N],n,val[N],rez[N];
int calc(int x)
{
if(!val[x])
{
return 0;
}
if(rez[x]==-1)
{
int t=val[x],rezi=x;
while(t && tata[rezi])
{
rezi=tata[rezi];
t--;
}
rez[x]=calc(rezi);
}
return rez[x]+1;
}
int main()
{
f>>n;
for(int i=1;i<=n;i++)
{
f>>val[i];
rez[i]=-1;
}
for(int i=1;i<n;i++)
{
int x,y;
f>>x>>y;
tata[y]=x;
}
for(int i=1;i<=n;i++)
{
g<<calc(i)<<" ";
}
return 0;
}