Cod sursa(job #2123080)

Utilizator giotoPopescu Ioan gioto Data 5 februarie 2018 19:35:44
Problema Cerere Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.9 kb
#include <cstdio>
#include <vector>
using namespace std;

int n;
int H[100005], NR[100005], k[100005];
bool f[100005], gr[100005];
vector <int> v[100005];
inline void dfs(int nod, int lev){
    for(auto it : v[nod]){
        if(f[it]) continue ;
        f[it] = 1;
        H[lev] = it;
        if(k[it] != 0)
        NR[it] = NR[H[lev - k[it]]] + 1;
        dfs(it, lev + 1);
    }
}
int main()
{
    freopen("cerere.in", "r", stdin);
    freopen("cerere.out", "w", stdout);
    scanf("%d", &n);
    for(int i = 1; i <= n ; ++i)
        scanf("%d", &k[i]);
    int x, y;
    for(int i = 1; i <= n ; ++i){
        scanf("%d%d", &x, &y);
        v[x].push_back(y);
        gr[y] = 1;
    }
    int w = 0;
    for(int i = 1; i <= n ; ++i)
        if(gr[i] == 0) w = i;
    f[w] = 1; H[1] = w;
    dfs(w, 2);
    for(int i = 1; i <= n ; ++i)
        printf("%d ", NR[i]);
    return 0;
}