Pagini recente » Cod sursa (job #1364429) | Profil MalinaC | Cod sursa (job #777853) | Cod sursa (job #1855512) | Cod sursa (job #1013162)
#include<fstream>
#include<vector>
#define MAXN 100001
using namespace std;
int n,back[MAXN],sol[MAXN],p[MAXN],st[MAXN],u;
vector<int>v[MAXN];
ifstream fin("cerere.in");
ofstream fout("cerere.out");
void read()
{
int x,y;
fin>>n;
for(int i=1;i<=n;i++)
fin>>back[i];
for(int i=1;i<n;i++)
{
fin>>x>>y;
v[x].push_back(y);
p[y]=x;
}
}
void dfs(int nod)
{
if(back[nod]!=0)
sol[nod]=sol[st[u-back[nod]]]+1;
for(int i=0;i<v[nod].size();i++)
{
st[++u]=v[nod][i];
dfs(v[nod][i]);
}
u--;
}
void print()
{
for(int i=1;i<=n;i++)
{
fout<<sol[i]<<" ";
}
}
int main()
{
read();
for(int i=1;i<=n;i++)
{
if(p[i]==0)
{
st[1]=i;
u=1;
dfs(i);
break;
}
}
print();
return 0;
}