Pagini recente » Cod sursa (job #2111477) | Cod sursa (job #2296030) | Cod sursa (job #2129270) | Cod sursa (job #1349636) | Cod sursa (job #926991)
Cod sursa(job #926991)
#include<fstream>
#include<algorithm>
#define NMAX 100005
using namespace std;
int n,v[NMAX],TT[NMAX],A[NMAX];
void read()
{
ifstream fin("cerere.in");
fin>>n;
int i,x,y;
for(i=1;i<=n;i++)
fin>>v[i];
for(i=1;i<n;i++)
{
fin>>x>>y;
TT[y]=x;
}
fin.close();
}
void solve()
{
for(int i=1,x;i<=n;i++)
if(v[i])
{
x=i;
int k=0;
while(v[x])
{
for(int i=x,aux=v[i];aux;aux--)
x=TT[x];
k++;
}
A[i]=k;
}
}
void print()
{
ofstream fout("cerere.out");
for(int i=1;i<=n;i++)
fout<<A[i]<<' ';
fout.close();
}
int main()
{
read();
solve();
print();
return 0;
}