Cod sursa(job #2778018)

Utilizator VipioanaMirea Oana Teodora Vipioana Data 27 septembrie 2021 15:19:49
Problema Cerere Scor 40
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.72 kb
#include <bits/stdc++.h>
#define N 100001

using namespace std;
ifstream f("cerere.in");
ofstream g("cerere.out");
int n,str[N],nr,t,k;
vector <int> a[N];

void dfs(int x, int t)
{
    if(t==k){
       nr++;
       if(!str[x])
        return;
       else
        k=str[x], t=0;
    }
    for(auto j:a[x])
        dfs(j,t+1);
}
int main()
{
    f>>n;
    for(int i=1; i<=n; i++)
        f>>str[i];
    for(int i=1; i<=n-1; i++){
        int x,y;
        f>>x>>y;
        a[y].push_back(x);
    }
    for(int i=1; i<=n; i++){
        if(!str[i])
            g<<"0 ";
        else{
            nr=0;
            k=str[i];
            dfs(i,0);
            g<<nr<<" ";
        }
    }
    return 0;
}