Cod sursa(job #2641735)

Utilizator cyg_vladioanBirsan Vlad cyg_vladioan Data 12 august 2020 16:43:51
Problema Cerere Scor 65
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.08 kb
#include <cstdio>
#include <vector>
using namespace std;
const int NMAX = 100000;
vector <int> G[NMAX + 5] , T[NMAX + 5];
int st[NMAX + 5] , s[NMAX + 5] , ans[NMAX + 5] , f[NMAX + 5] , root;
void dfs1(int u , int lvl)
{
    st[lvl] = u;
    if(s[u] != 0)
        T[st[lvl - s[u]]].push_back(u);
    else
        T[0].push_back(u);
    for(int j = 0 ; j < G[u].size() ; j ++)
        dfs1(G[u][j] , lvl + 1);
}
void dfs2(int u , int lvl)
{
    ans[u] = lvl - 1;
    for(int j = 0 ; j < T[u].size() ; j ++)
        dfs2(T[u][j] , lvl + 1);
}
int main()
{
    freopen("cerere.in" , "r" , stdin);
    freopen("cerere.out" , "w" , stdout);
    int n , i , x , y;
    scanf("%d" , &n);
    for(i = 1 ; i <= n ; i ++)
        scanf("%d" , &s[i]);
    for(i = 1 ; i < n ; i ++)
    {
        scanf("%d%d" , &x , &y);
        G[x].push_back(y);
        f[y] = 1;
    }
    for(i = 1 ; i <= n && !root ; i ++)
        if(!f[i])
            root = i;
    dfs1(root , 0);
    dfs2(0 , 0);
    for(i = 1 ; i <= n ; i ++)
        printf("%d " , ans[i]);
    return 0;
}