Cod sursa(job #2778018)
Utilizator | 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;
}