Pagini recente » Cod sursa (job #1501686) | Cod sursa (job #291956) | Cod sursa (job #2476647) | Cod sursa (job #442818) | Cod sursa (job #2595627)
#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]<<" ";
}