Cod sursa(job #2595623)

Utilizator denmirceaBrasoveanu Mircea denmircea Data 8 aprilie 2020 00:46:50
Problema Cerere Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.84 kb
#include <bits/stdc++.h>
#define dim 100005
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
int val[dim],r[dim],curr[dim];
bitset<dim> fr,nr;
int niv,n,x,y,i;
vector <int> L[dim];
void dfs(int nod)
{
    if(val[nod])
        r[nod]=r[curr[niv-val[nod]]]+1;
    curr[niv]=nod;
    fr[nod]=1;
    for(auto it:L[nod])
    {
        if(fr[it]==0){
            niv++;
            dfs(it);
            niv--;
        }
    }
}
int main()
{
    ios_base::sync_with_stdio(false);
    fin>>n;
    for(i=1; i<=n; i++)
        fin>>val[i];
    for(i=1; i<n; i++)
    {
        fin>>x>>y;
        L[x].push_back(y);
        nr[y]=1;
    }
    for(i=1; i<=n; i++)
    {
        if(nr[i])
            continue;
        niv=1;
        dfs(i);
    }
    for(i=1;i<=n;i++)
        fout<<r[i]<<" ";
}