Cod sursa(job #3199063)

Utilizator Razvan23Razvan Mosanu Razvan23 Data 31 ianuarie 2024 16:50:46
Problema Cerere Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.81 kb
#include <bits/stdc++.h>
using namespace std;

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

vector<int> L[100005];
int d[100005], nr[100005], st[100005];
bitset<100005> g;
int n, rad, m;

void DFS(int k)
{
    st[++m] = k;
    if(d[k]) nr[k] = nr[st[m - d[k]]] + 1;
    for(auto w : L[k])
        DFS(w);
    m--;
}

int main()
{
    ios_base::sync_with_stdio(0);
    fin.tie(0);
    fout.tie(0);
    int i, x, y;
    fin >> n;
    for(i=1; i<=n; i++)
        fin >> d[i];
    for(i=1; i<n; i++)
    {
        fin >> x >> y;
        L[x].push_back(y);
        g[y] == 1;
    }
    for(i=1; i<=n && rad == 0; i++)
        if(g[i] == 0) rad = i;
    DFS(rad);
    for(i=1;i<=n;i++)
        fout << nr[i] << " ";
    fin.close();
    fout.close();
    return 0;
}