Pagini recente » Cod sursa (job #1456842) | Cod sursa (job #56807) | Cod sursa (job #1317942) | Cod sursa (job #2277243) | Cod sursa (job #2144292)
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
vector <int> a[100001];
int n,m,v[100001],rasp[100001],st[100001],vf,t[100001];
bool viz[100001];
ifstream in("cerere.in");
ofstream out("cerere.out");
void dfs(int x)
{
st[++vf]=x;
if(v[x]==0)
rasp[x]=0;
else
rasp[x]=rasp[st[vf-v[x]]]+1;
int y;
viz[x]=true;
for(unsigned i=0;i<a[x].size();i++)
{
y=a[x][i];
if(!viz[y])
dfs(y);
}
vf--;
}
int main()
{
int x,y,j,i;
in>>n;
for(j=1;j<=n;j++)
in>>v[j];
for(j=1;j<n;j++)
{
in>>x>>y;
t[y]=x;
a[x].push_back(y);
}
for(i=1;i<=n;i++)
if(t[i]==0)
{
dfs(i);
break;
}
for(j=1;j<=n;j++)
out<<rasp[j]<<' ';
return 0;
}