Cod sursa(job #2641737)

Utilizator cyg_vladioanBirsan Vlad cyg_vladioan Data 12 august 2020 16:48:20
Problema Cerere Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.66 kb
#include <cstdio>
#include <vector>
#include <cctype>
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);
}
char buffer[1 << 17];
int sz = (1 << 17) , crs = (1 << 17);
void get_int(int &n)
{
    for( ; crs < sz && !isdigit(buffer[crs]) ; crs ++);
    if(crs == sz)
    {
        fread(buffer , sz , 1 , stdin);
        for(crs = 0 ; crs < sz && !isdigit(buffer[crs]) ; crs ++);
    }
    n = 0;
    for( ; crs < sz && isdigit(buffer[crs]) ; crs ++)
        n = n * 10 + buffer[crs] - '0';
    if(crs == sz)
    {
        fread(buffer , sz , 1 , stdin);
        for(crs = 0 ; crs < sz && isdigit(buffer[crs]) ; crs ++)
            n = n * 10 + buffer[crs] - '0';
    }
}
int main()
{
    freopen("cerere.in" , "r" , stdin);
    freopen("cerere.out" , "w" , stdout);
    int n , i , x , y;
    get_int(n);
    for(i = 1 ; i <= n ; i ++)
        get_int(s[i]);
    for(i = 1 ; i < n ; i ++)
    {
        get_int(x);
        get_int(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;
}