Pagini recente » Cod sursa (job #349052) | Cod sursa (job #2816411) | Cod sursa (job #58272) | Cod sursa (job #2703210) | Cod sursa (job #2542103)
#include <bits/stdc++.h>
using namespace std;
//ifstream f("memory001.in");
//ofstream g("memory001.out");
ifstream f("cerere.in");
ofstream g("cerere.out");
const int dim = 100001;
int n;
int k[dim];
vector <int> sons[dim];
int parent[dim];
int dist[dim];
vector <int> nodes;
void dfs(int node){
nodes.push_back(node);
if(k[node] > 0){
dist[node] = dist[nodes[nodes.size() - k[node] - 1]] + 1;
}
for(int it: sons[node])
dfs(it);
nodes.pop_back();
}
int main()
{
int i, a, b;
f >> n;
for(i = 1; i <= n; ++i){
f >> k[i];
}
for(i = 1; i < n; ++i){
f >> a >> b;
sons[a].push_back(b);
parent[b] = a;
}
int root;
for(i = 1; i <= n; ++i)
if(parent[i] == 0){
root = i;
break;
}
dfs(root);
for(i = 1; i <= n; ++i)
g << dist[i] << " ";
return 0;
}