Pagini recente » Cod sursa (job #2875103) | Cod sursa (job #705693) | Cod sursa (job #1418764) | Cod sursa (job #295522) | Cod sursa (job #2146068)
#include <iostream>
#include <cstdio>
#include <vector>
#define MAXN 100001
using namespace std;
vector <int> v[MAXN];
int stramos[MAXN],st[MAXN],d[MAXN],vf;
bool viz[MAXN];
void dfs(int x)
{
int y;
st[vf++]=x;
for(unsigned int i=0;i<v[x].size();i++)
{
y=v[x][i];
if(stramos[y])
d[y]=d[st[vf-stramos[y]]]+1;
dfs(y);
}
vf--;
}
int main()
{
FILE *fin,*fout;
fin=fopen("cerere.in","r");
fout=fopen("cerere.out","w");
int n,x,y;
fscanf(fin,"%d",&n);
for(int i=1;i<=n;i++)
fscanf(fin,"%d",&stramos[i]);
for(int i=1;i<n;i++)
{
fscanf(fin,"%d%d",&x,&y);
v[x].push_back(y);
viz[y]=true;
}
for(int i=1;i<=n;i++)
if(!viz[i])
{
x=i;
break;
}
dfs(x);
for(int i=1;i<=n;i++)
fprintf(fout,"%d ",d[i]);
fclose(fin);
fclose(fout);
return 0;
}