Cod sursa(job #2429378)

Utilizator IulianOleniucIulian Oleniuc IulianOleniuc Data 9 iunie 2019 13:39:02
Problema Cerere Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.79 kb
#include <vector>
#include <fstream>

#define NMAX 100010
using std::vector;

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

int n;
int anc[NMAX];
vector<int> ad[NMAX];

int root;
int sol[NMAX];

int top;
int st[NMAX];

void dfs(int node) {
    st[++top] = node;
    sol[node] = sol[st[top - anc[node]]] + 1;

    for (int nghb : ad[node])
        dfs(nghb);
    top--;
}

int main() {
    fin >> n;
    for (int i = 1; i <= n; i++)
        fin >> anc[i];

    root = 1LL * n * (n + 1) / 2;
    for (int i = 1; i < n; i++) {
        int x, y; fin >> x >> y;
        ad[x].push_back(y);
        root -= y;
    }

    dfs(root);
    for (int i = 1; i <= n; i++)
        fout << sol[i] - 1 << ' ';
    fout << '\n';

    fout.close();
    return 0;
}