Pagini recente » Cod sursa (job #902511) | Cod sursa (job #3153890) | Sandbox (cutiuţa cu năsip) | Cod sursa (job #1575642) | Cod sursa (job #1665894)
#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], L[20][Nmax], log2[Nmax];
vector <int> G[Nmax];
int Stramos(int q, int p)
{
while(p)
{
int k = log2[p];
q = L[k][q];
p = p - (1<<k);
}
return q;
}
void Precalculate()
{
for(int i = 2; i <= n; i++) log2[i] = log2[i/2] + 1;
for(int i = 1; (1<<i) <= n; i++)
for(int j = 1; j <= n; j++) L[i][j] = L[i-1][L[i-1][j]];
}
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);
L[0][y] = x;
}
Precalculate();
for(int i = 1; i <= n; i++)
{
if(L[0][i] == 0)
{
DFS(i);
i = n+1;
}
}
for(int i = 1; i <= n; i++)
{
g<<sol[i]<<' ';
}
return 0;
}