Cod sursa(job #1829319)

Utilizator Cezar_MihalceaCezar Mihalcea Cezar_Mihalcea Data 14 decembrie 2016 19:40:08
Problema Cerere Scor 5
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.61 kb
#include<fstream>
using namespace std;

ifstream f("cerere.in");
ofstream g("cerere.out");

int pred[100001], v[100001], sol[100001], n, nr;

void adauga(int x, int y)
{
    pred[y]=x;
}

int main()
{
    int i,j,x,y, k;
    f>>n;
    for(i=1;i<=n;i++)
        f>>v[i];
    for(i=1;i<=n-1;i++)
    {
        f>>x>>y;
        adauga(x,y);
    }
    for(i=1;i<=n;i++)
    {
        if(v[i] == 0)
            sol[i] = 0;
        else
        {
            k=i;
            for(j=1;j<=v[i];j++)
                k=pred[k];
            sol[i]=sol[k] + 1;
        }
    }
    for(i=1;i<=n;i++)
        g<<sol[i]<<" ";
    return 0;
}