Cod sursa(job #2316135)

Utilizator TheNextGenerationAyy LMAO TheNextGeneration Data 11 ianuarie 2019 11:13:49
Problema Cerere Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.87 kb
#include <bits/stdc++.h>

using namespace std;

const int N = 1e5+1;

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

int val[N],rez[N],lvl[N],st[N];
bool a[N];
vector<int> v[N];

void dfs(int x)
{
    for (auto it: v[x])
        if (!lvl[it])
        {
            lvl[it] = lvl[x]+1;
            st[lvl[it]] = it;
            if (val[it])
                rez[it] = rez[st[lvl[it]-val[it]]]+1;
            dfs(it);
        }
}

int main()
{
    int n;
    in >> n;
    for (int i = 1; i<=n; i++)
        in >> val[i];
    for (int i = 1; i<n; i++)
    {
        int x,y;
        in >> x >> y;
        v[x].push_back(y);
        a[y] = 1;
    }
    int root = 0;
    for (int i = 1; i<=n; i++)
        if (!a[i])
            root = i;
    st[1] = root;
    dfs(root);
    for (int i = 1; i<=n; i++)
        out << rez[i] << " ";
}