Pagini recente » Cod sursa (job #539153) | Cod sursa (job #1788326) | Cod sursa (job #2573158) | Cod sursa (job #2421250) | Cod sursa (job #2778018)
#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;
}