Pagini recente » Cod sursa (job #992371) | Cod sursa (job #1767445) | Cod sursa (job #1657408) | Cod sursa (job #534397) | Cod sursa (job #1665896)
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
ifstream f("cerere.in");
ofstream g("cerere.out");
const int Nmax = 100005;
int n, k[Nmax], Use[Nmax], sol[Nmax],tata[Nmax] , log2[Nmax];
vector <int> G[Nmax];
int Stramos(int q, int p)
{
while(p--)
{
q = tata[q];
}
return q;
}
void DFS(int nod)
{
Use[nod] = 1;
for(int i = 0; i < (int)G[nod].size(); i++)
{
int vecin = G[nod][i];
int stramos = Stramos(vecin, k[vecin]);
if(vecin != stramos) sol[vecin] = sol[stramos] + 1;
DFS(vecin);
}
}
int main()
{
f>>n;
for(int i = 1; i <= n; i++) f>>k[i];
for(int i = 1; i < n; i++)
{
int x,y;
f>>x>>y;
G[x].push_back(y);
tata[y] = x;
}
for(int i = 1; i <= n; i++)
{
if(tata[i] == 0)
{
DFS(i);
i = n+1;
}
}
for(int i = 1; i <= n; i++)
{
g<<sol[i]<<' ';
}
return 0;
}