Cod sursa(job #1700850)

Utilizator RaduMirceaAndreiRadu Mircea Andrei RaduMirceaAndrei Data 11 mai 2016 15:14:50
Problema Cerere Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.91 kb
# include <cstdio>
# include <vector>
# define DIM 100010
using namespace std;
vector <int> Lista[DIM];
int Marcat[DIM],d[DIM],sol[DIM],v[DIM],h[DIM],n,i,x,y;
void dfs(int nc,int val){
    int nv;
    Marcat[nc]=1;
    if(v[nc]==0)
        d[val]=0;
    else
        d[val]=d[val-v[nc]]+1;
    sol[nc]=d[val];
    for(int i=0;i<Lista[nc].size();i++){
        nv=Lista[nc][i];
        if(Marcat[nv]==0)
            dfs(nv,val+1);
    }
}
int main () {
    freopen("cerere.in","r",stdin);
    freopen("cerere.out","w",stdout);
    scanf("%d",&n);
    for(i=1;i<=n;i++)
        scanf("%d",&v[i]);
    for(i=1;i<=n-1;i++){
        scanf("%d%d",&x,&y);
        Lista[x].push_back(y);
        h[y]=1;
    }
    for(i=1;i<=n;i++){
        if(h[i]==0){
            dfs(i,1);
            break;
        }
    }
    for(i=1;i<=n;i++)
        printf("%d ",sol[i]);
    printf("\n");
    return 0;
}