Cod sursa(job #1028426)

Utilizator george_stelianChichirim George george_stelian Data 14 noiembrie 2013 00:59:23
Problema Cerere Scor 40
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.82 kb
#include <cstdio>
#include <vector>

using namespace std;

vector<int>v[100000];
int d[100000],v2[100000],sol[100000];
char v1[100000];
int n,i,x,y;

void adancime(int x,int y)
{
    int i,lim;
    v2[y]=x;
    if(d[x]==0) sol[x]=0;
    else sol[x]=sol[v2[y-d[x]]]+1;
    lim=v[x].size();
    for(i=0;i<lim;i++) adancime(v[x][i],y+1);
}

int main()
{
    freopen("cerere.in", "r", stdin);
    freopen("cerere.out", "w", stdout);
    scanf("%d",&n);
    for(i=1;i<=n;i++) scanf("%d",&d[i]);
    for(i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        v[x].push_back(y);
        v1[y]=1;
    }
    for(i=1;i<=n;i++)
    {
        if(v1[i]==0)
        {
            adancime(i,1);
        }
    }
    for(i=1;i<=n;i++) printf("%d ",sol[i]);
    fclose(stdin);fclose(stdout);
    return 0;
}