Pagini recente » Cod sursa (job #2681897) | Cod sursa (job #2178198) | Cod sursa (job #1324329) | Cod sursa (job #907683) | Cod sursa (job #2595625)
#include <ios>
#include <fstream>
#include <vector>
#define dim 100005
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
int val[dim],r[dim],curr[dim];
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;
for(auto it:L[nod])
{
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);
}
for(i=1; i<=n; i++)
{
if(L[i].size()==0)
continue;
niv=1;
dfs(i);
}
for(i=1;i<=n;i++)
fout<<r[i]<<" ";
}