Pagini recente » Cod sursa (job #1729301) | Cod sursa (job #1611588) | Cod sursa (job #1346588) | Cod sursa (job #1752207) | Cod sursa (job #1744757)
#include <iostream>
#include <fstream>
#include <vector>
#include <bitset>
using namespace std;
const int nmax=100005;
int k[nmax],st[nmax],nr[nmax],a,b,top,n,i,node;
vector<int> v[nmax];
bitset<nmax> noroot;
void dfs(int x)
{
top++;
st[top]=x;
if(k[x]==0)nr[x]=0;
else nr[x]=nr[st[top-k[x]]]+1;
for(int i=0;i<v[x].size();i++)
dfs(v[x][i]);
top--;
}
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;
v[a].push_back(b);
noroot[b]=1;
}
for(i=1;i<=n;i++)
if(noroot[i]==0)
{
node=i;
break;
}
dfs(node);
for(i=1;i<=n;i++)
g<<nr[i]<<' ';
return 0;
}