Pagini recente » Cod sursa (job #1333979) | Cod sursa (job #3219982) | Cod sursa (job #97428) | Cod sursa (job #2235941) | Cod sursa (job #588717)
Cod sursa(job #588717)
#include <fstream>
#include <vector>
using namespace std;
const int N=100002,LG=18,inf=1<<30;
int a[N][LG],v[N],L[N],rez[N],n;
vector<int> fii[N];
ifstream in("cerere.in");
ofstream out("cerere.out");
int stramos(int x,int y)
{
if (y==(1<<L[y]))
return a[L[y]][x];
return stramos(a[L[y]][x],y-(1<<L[y]));
}
void dfs(int x)
{
if (!v[x])
rez[x]=0;
else
rez[x]=rez[v[x]]+1;
for (vector<int>:: iterator i=fii[x].begin();i!=fii[x].end();i++)
dfs(*i);
}
void print(int v[])
{
for (int i=1;i<=n;i++)
out<<v[i]<<" ";
out<<"\n";
}
int main()
{
int i,j,x,y;
in>>n;
L[0]=-1;
for (i=0;i<LG-1;i++)
L[1<<i]=i;
for (i=1;i<=n;i++)
{
in>>v[i];
if (!L[i])
L[i]=L[i-1];
}
L[1]=0;
for (i=1;i<n;i++)
{
in>>x>>y;
a[0][y]=x;
fii[x].push_back(y);
}
for (j=1;j<LG;j++)
for (i=1;i<=n;i++)
a[j][i]=a[j-1][a[j-1][i]];
for (i=1;i<=n;i++)
if (v[i])
v[i]=stramos(i,v[i]);
for (i=1;i<=n;i++)
if (!a[0][y])
{
dfs(i);
rez[i]=0;
}
print(rez);
return 0;
}