Pagini recente » Cod sursa (job #2267983) | Cod sursa (job #398620) | Cod sursa (job #2726532) | Cod sursa (job #1526772) | Cod sursa (job #1835279)
#include <fstream>
#include <vector>
#include <stack>
using namespace std;
vector <int> v[100005];
int n,i,a,b,rad,fiu,vf,k[100005],rez[100005],s[100005],sf[100005];
char viz[100005];
int main()
{
ifstream f("cerere.in");
ofstream g("cerere.out");
f>>n;
for(i=1; i<=n; i++)
f>>k[i];
for(i=1; i<=n-1; i++)
{
f>>a>>b;
viz[b]=1;
v[a].push_back(b);
}
rad=1;
while(viz[rad]) rad++;
s[1]=rad;
vf=1;
sf[1]=0;
while(vf)
{
rad=s[vf];
fiu=sf[vf];
if(k[rad])rez[rad]=rez[s[vf-k[rad]]]+1;
if(fiu<v[rad].size())
{
s[++vf]=v[rad][fiu];
sf[vf]=0;
sf[vf-1]=fiu+1;
}
else vf--;
}
for(i=1; i<=n; i++) g<<rez[i]<<' ';
f.close(); g.close();
return 0;
}