Cod sursa(job #2883378)

Utilizator BorodiBorodi Bogdan Borodi Data 1 aprilie 2022 14:30:24
Problema Cerere Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.92 kb
	
#include <bits/stdc++.h>
#define Nmax 100001
using namespace std;

ifstream fin("cerere.in");
ofstream fout("cerere.out");

typedef vector < pair < int, int > > VIP;
typedef vector < int > VI;

const int oo = 1 << 28;

int n, dad, x, y;
int G[Nmax], Ans[Nmax], Arb[Nmax], In[Nmax], dp[Nmax];
VI V[Nmax];

void DFS(int vertex, int lv){
    Arb[lv] = vertex;
    dp[vertex] = dp[Arb[lv - G[vertex]]] + 1;

    for(int new_vertex : V[vertex])
        DFS(new_vertex, lv + 1);
}

int main() {
    fin >> n;

    for(int i = 1; i <= n; ++i)
        fin >> G[i];    

    for(int i = 1; i <= n - 1; ++i){
        fin >> x >> y;

        V[x].push_back(y);
        In[y]++;
    }

    for(int i = 1; i <= n - 1; ++i)
        if(In[i] == 0){
            dad = i;
            break;
        }

    DFS(dad, 1);

    for(int i = 1; i <= n; ++i)
        fout << dp[i] - 1 << ' ' ;

    return 0;   
}