Pagini recente » Cod sursa (job #1891724) | Cod sursa (job #1104026) | Cod sursa (job #147481) | Cod sursa (job #203) | Cod sursa (job #3272174)
#include <bits/stdc++.h>
using namespace std;
vector<int> adj[100005];
int v[100005];
int st[100005], l;
int f[100005];
bool vis[100005];
bool vis1[100005];
void dfs(int node){
vis[node]=1;
st[++l]=node;
f[node]=st[l-v[node]];
for(auto i : adj[node]){
if(vis[i]==0)
dfs(i);
}
--l;
}
int ras[100005];
int main(){
ifstream cin ("cerere.in");
ofstream cout ("cerere.out");
int n;
cin >> n;
for(int i=1; i<=n; i++)
cin >> v[i];
int a, b;
for(int i=1; i<=n-1; i++){
cin >> a >> b;
adj[a].push_back(b);
vis1[b]=1;
}
for(int i=1; i<=n; i++){
if(vis1[i]==0){
dfs(i);
break;
}
}
for(int i=1; i<=n; i++)
ras[i]=-1;
for(int i=1; i<=n; i++){
if(v[i]==0)
ras[i]=0;
}
for(int i=1; i<=n; i++){
if(v[i]!=0){
int poz=i, cnt=0;
while(ras[poz]==-1){
poz=f[poz];
cnt++;
}
cnt+=ras[poz];
poz=i;
while(ras[poz]==-1){
ras[poz]=cnt;
poz=f[poz];
cnt--;
}
}
}
for(int i=1; i<=n; i++)
cout << ras[i] << ' ';
return 0;
}