Pagini recente » Cod sursa (job #2477004) | Cod sursa (job #2157196) | Cod sursa (job #1192485) | Cod sursa (job #1326388) | Cod sursa (job #2593434)
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
ifstream f("cerere.in");
ofstream g("cerere.out");
vector <int> q[100010];
int a[100010],b[100010],t[100010],rad,n,i,j,k,nr;
bool viz[100010];
void dfs(int i)
{ if(a[i]!=0)
{
b[i]=b[t[nr-a[i]+1]]+1;//cout<<i<<" "<<y<<" "<<b[i]<<" "<<b[y]<<"\n";
}
t[++nr]=i;
for(vector <int> ::iterator it=q[i].begin();it<q[i].end();it++)
dfs(*it);
nr--;
}
int main()
{ f>>n;
for(k=1;k<=n;k++)
f>>a[k];
for(k=1;k<n;k++)
{f>>i>>j;viz[j]=0;
q[i].push_back(j);
}
for(i=1;i<=n;i++)
if(viz[i]==0) {rad=i;break;}
dfs(rad);
for(i=1;i<=n;i++)
g<<b[i]<<" ";
return 0;
}