Cod sursa(job #1054187)

Utilizator usermeBogdan Cretu userme Data 13 decembrie 2013 14:30:54
Problema Cerere Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.74 kb
#include <cstdio>
#include <vector>

using namespace std;

FILE*f=fopen("cerere.in","r");
FILE*h=fopen("cerere.out","w");

int s[100001],t[100001],n,e[100001],k,r[100001];

vector<int> v[100001];

inline void dfs( int x ){
    s[++k]=x;
    if ( t[x] )
        r[x]=r[s[k-t[x]]]+1;
    for ( int i=0;i<v[x].size();++i )
        dfs(v[x][i]);
    --k;
}

int main()
{
    fscanf(f,"%d",&n);
    for ( int i=1;i<=n;++i )
        fscanf(f,"%d",&t[i]);
    for ( int i=1;i<n;++i ){
        int a,b;
        fscanf(f,"%d%d",&a,&b);
        v[a].push_back(b);
        e[b]=1;
    }
    for ( int i=1;i<=n;++i )
        if ( e[i]==0 ){dfs(i);break;}
    for ( int i=1;i<=n;++i )
        fprintf(h,"%d ",r[i]);
    return 0;
}