Pagini recente » Cod sursa (job #2246970) | Cod sursa (job #1987715) | Cod sursa (job #1454871) | Cod sursa (job #16771) | Cod sursa (job #2139192)
#include<fstream>
#include<vector>
using namespace std;
ifstream f("cerere.in");
ofstream g("cerere.out");
int n, k[100002], d[100002], s[100002], h, rad;
bool viz[100002];
vector<int> l[100002];
void dfs(int nod)
{
s[++h]=nod;
viz[nod]=1;
if(k[nod]!=0)
d[nod]=d[s[h-k[nod]]]+1;
for(int i=0;i<l[nod].size();i++)
if(viz[l[nod][i]]==0)
dfs(l[nod][i]);
h--;
}
int main()
{
f>>n;
for(int i=1;i<=n;i++)
{
f>>k[i];
if(rad==0 && k[i]==0)
rad=i;
}
for(int i=1;i<n;i++)
{
int x, y;
f>>x>>y;
l[x].push_back(y);
l[y].push_back(x);
}
dfs(rad);
for(int i=1;i<=n;i++)
g<<d[i]<<" ";
return 0;
}