Cod sursa(job #2973161)

Utilizator NiffSniffCojocaru Calin Marcu NiffSniff Data 31 ianuarie 2023 09:23:35
Problema Cerere Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.94 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream cin ("cerere.in");
ofstream cout ("cerere.out");
const int N = 100000;
int help[N+1],n,timp[N+1], subord[N+1];
vector <int> L[N+1], coada;
void afisare()
{
    for (int i : coada)
        cout << i << ' ';
    cout << endl;
}
void dfs(int x)
{
    coada.push_back(x);
    if (help[x] == 0)
        timp[x] = 0;
    else
    {
        timp[x] = timp[coada[coada.size()-1-help[x]]] + 1;
    }
    for (int y : L[x])
    {
        dfs(y);
    }
    coada.pop_back();
}
int main()
{
    int x,y,inteleptu(1);
    cin >> n;
    for (int i=1; i<=n; i++)
        cin >> help[i];
    for (int i=1; i<n; i++)
    {
        cin >> x >> y;
        L[x].push_back(y);
        subord[y] = x;
    }
    while (subord[inteleptu])
    {
        inteleptu = subord[inteleptu];
    }
    dfs(inteleptu);
    for (int i=1; i<=n; i++)
        cout << timp[i] << ' ';
}