Cod sursa(job #3139586)

Utilizator Mihai_PopescuMihai Popescu Mihai_Popescu Data 30 iunie 2023 10:26:46
Problema Cerere Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.76 kb
#include <fstream>
#include <vector>
using namespace std;

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

#define NMAX 100005

vector<int> G[NMAX];

int k[NMAX], viz[NMAX];
int rez[NMAX], lant[NMAX];

void dfs(int nod, int niv)
{
    viz[nod] = 1;
    lant[niv] = nod;

    if (k[nod])
        rez[nod] = rez[lant[niv - k[nod]]] + 1;

    for (auto it : G[nod])
        if (!viz[it])
            dfs(it, niv + 1);
}

int main()
{
    int n;
    fin >> n;

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

    for (int i = 1; i < n; ++i)
    {
        int a, b;
        fin >> a >> b;
        G[a].push_back(b);
    }

    dfs(1, 1);

    for (int i = 1; i <= n; ++i)
        fout << rez[i] << ' ';
    return 0;
}