Pagini recente » Cod sursa (job #3003602) | Cod sursa (job #1541235) | Cod sursa (job #1812583) | Cod sursa (job #2928578) | Cod sursa (job #926521)
Cod sursa(job #926521)
#include<fstream>
#include<algorithm>
#define NMAX 100005
using namespace std;
int n,v[NMAX],str[18][NMAX],A[NMAX];
void read()
{
ifstream fin("cerere.in");
fin>>n;
int i,x,y;
for(i=1;i<=n;i++)
fin>>v[i];
for(i=1;i<n;i++)
{
fin>>x>>y;
str[0][y]=x;
}
fin.close();
}
inline void stramosi()
{
for(int i=1;i<=n;i++)
for(int j=1;(1<<j)<=n;j++)
str[j][i]=str[j-1][str[j-1][i]];
}
inline int query(int x,int y)
{
if(!y)
return x;
int i;
for(i=0;(1<<i)<=y;i++);
i--;
y-=(1<<i);
x=str[i][x];
return query(x,y);
}
void solve()
{
stramosi();
for(int i=1,x;i<=n;i++)
if(v[i])
{
x=query(i,v[i]);
A[i]=A[x]+1;
}
}
void print()
{
ofstream fout("cerere.out");
for(int i=1;i<=n;i++)
fout<<A[i]<<' ';
fout.close();
}
int main()
{
read();
solve();
print();
return 0;
}