Cod sursa(job #3148427)

Utilizator andrei1807Andrei andrei1807 Data 1 septembrie 2023 11:26:01
Problema Cerere Scor 50
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.94 kb
#include <iostream>
#include <fstream>
#include <vector>

using namespace std;

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

const int NMAX = 100003;

vector<int> g[NMAX];
int tata[NMAX], k[NMAX], s[NMAX];
bool vis[NMAX];
int radacina = 1;

void dfs(int nod) {
    vis[nod] = true;

    if (k[nod] != 0) {
        int stramos = nod;
        for (int i = 1; i <= k[nod]; i++)
            stramos = tata[stramos];

        s[nod] = s[stramos] + 1;
    }

    for (auto fiu : g[nod]) {
        if (!vis[fiu])
            dfs(fiu);
    }
}

int main() {

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

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

        tata[y] = x;
        g[x].push_back(y);
    }

    while (tata[radacina] != 0) {
        radacina = tata[radacina];
    }

    dfs(radacina);
    for (int i = 1; i <= n; i++) {
        fout << s[i] << ' ';
    }

    return 0;
}