Cod sursa(job #2673114)

Utilizator BereaBerendea Andrei Berea Data 15 noiembrie 2020 19:26:01
Problema Cerere Scor 40
Compilator cpp-64 Status done
Runda ursus_retro_fara_alcool Marime 0.61 kb
//#include <iostream>
#include <fstream>

using namespace std;
const int maxn=1e5;
int n,i,j,x,y,c;
int v[maxn],t[maxn];

ifstream cin("cerere.in");
ofstream cout("cerere.out");

void da()
{
    c=0;
    y=i;
    x=v[y];
    while (x!=0)
    {
        for (j=1;j<=x;j++)
        {
            y=t[y];
        }
        c++;
        x=v[y];
    }
    cout<<c<<" ";
}

int main()
{
    cin>>n;
    for (i=1;i<=n;i++) cin>>v[i];
    for (i=1;i<n;i++)
    {
        cin>>x>>y;
        t[y]=x;
    }
    for (i=1;i<=n;i++)
    {
        if (v[i]==0) cout<<0<<" ";
        else da();
    }
}