Pagini recente » Cod sursa (job #2664121) | Cod sursa (job #5715) | Cod sursa (job #1598010) | Cod sursa (job #1142292) | Cod sursa (job #2781098)
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
ifstream in("cerere.in");
ofstream out("cerere.out");
int n, lantLun, r;
int stramos[100100], ans[100100], lant[100100];
bool fr[100100];
vector<vector<int>> adj;
void dfs(int node, int father)
{
lant[++lantLun] = node;
if(stramos[node] != 0)
{
int indStramos = lantLun - stramos[node];
int str = lant[indStramos];
ans[node] = ans[str] + 1;
}
for(auto it: adj[node])
{
if(it == father)
continue;
dfs(it, node);
}
lantLun--;
}
int main()
{
ios_base::sync_with_stdio(false);
in >> n;
for(int i = 1; i <= n; i++)
in >> stramos[i];
adj.resize(n + 5);
for(int i = 1; i < n; i++)
{
int x, y;
in >> x >> y;
adj[x].push_back(y);
adj[y].push_back(x);
fr[y] = 1;
}
for(int i = 1; i <= n && r == 0; i++)
if(fr[i] == 0)
r = i;
dfs(r, 0);
for(int i = 1; i <= n; i++)
out << ans[i] << ' ';
out << '\n';
return 0;
}