Cod sursa(job #2940547)

Utilizator carinamariaCarina Maria Viespescu carinamaria Data 15 noiembrie 2022 20:00:20
Problema Cerere Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.79 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream cin("cerere.in");
ofstream cout("cerere.out");
int i, j, n, m, x, y, nr, minim, r;
int maxim[100002], s[100002], v[100002], sol[100002], t[100002];
vector <int> L[100002];
void dfs(int nod, int niv){
    s[niv]=nod;
    if(v[nod]==0)
        sol[nod]=0;
    else
        sol[nod]=sol[s[niv-v[nod]]]+1;
    for(int i=0;i<L[nod].size();i++){
        dfs(L[nod][i], niv+1);
    }
}
int main() {
    cin>>n;
    for(i=1;i<=n;i++){
        cin>>v[i];
    }
    for(i=1;i<n;i++){
        cin>>x>>y;
        L[x].push_back(y);
        t[y]=x;
    }
    for(i=1;i<=n;i++){
        if(t[i]==0){
            r=i;
            break;
        }
    }
    dfs(r, 1);
    for(i=1;i<=n;i++)
    cout<<sol[i]<<" ";


}