Pagini recente » Cod sursa (job #862640) | Cod sursa (job #1694538) | Cod sursa (job #2437498) | Cod sursa (job #2573661) | Cod sursa (job #1068855)
#include <fstream>
#include <vector>
using namespace std;
vector<int> l[100005];
int n, i, a, b, x;
int k[100005], v[100005], P[100005];
bool ok[100005], o[100005];
void dfs(int x, int niv)
{
ok[x]=1;
P[niv]=x;
v[x]=v[P[niv-k[x]]]+1;
for(int i=0;i<l[x].size();i++)
{
int y=l[x][i];
if(!ok[y])
dfs(y, niv+1);
}
}
int main()
{
ifstream f("cerere.in");
ofstream g("cerere.out");
f>>n;
for(i=1;i<=n;i++)
{
f>>k[i];
v[i]=-1;
}
for(i=1;i<n;i++)
{
f>>a;f>>b;
l[a].push_back(b);
l[b].push_back(a);
o[b]=1;
}
for(i=1;i<=n;i++)
if(!o[i])
{
x=i;
break;
}
dfs(x,1);
for(i=1;i<=n;i++)
g<<v[i]<<" ";
return 0;
}