Cod sursa(job #2980742)

Utilizator rapidu36Victor Manz rapidu36 Data 16 februarie 2023 19:38:11
Problema Cerere Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.01 kb
#include <fstream>
#include <vector>

using namespace std;

const int N = 1e5;

vector <int> fii[N+1];
int t[N+1], stramos[N+1], drum[N+1], nr_c[N+1];

void dfs(int x, int nivel)
{
    drum[nivel] = x;
    if (stramos[x] != 0)
    {
        nr_c[x] = 1 + nr_c[drum[nivel - stramos[x]]];
    }
    for (auto y: fii[x])
    {
        dfs(y, nivel + 1);
    }
}

int radacina(int n)
{
    for (int i = 1; i <= n; i++)
    {
        if (t[i] == 0)
        {
            return i;
        }
    }
    return 0;
}

int main()
{
    ifstream in("cerere.in");
    ofstream out("cerere.out");
    int n;
    in >> n;
    for (int i = 1; i <= n; i++)
    {
        in >> stramos[i];
    }
    for (int i = 1; i < n; i++)
    {
        int x, y;
        in >> x >> y;
        t[y] = x;
        fii[x].push_back(y);
    }
    int r = radacina(n);
    dfs(r, 0);
    for (int i = 1; i <= n; i++)
    {
        out << nr_c[i] << " ";
    }
    in.close();
    out.close();
    return 0;
}