Pagini recente » Cod sursa (job #2296242) | Cod sursa (job #1946967) | Cod sursa (job #884390) | Cod sursa (job #1588585) | Cod sursa (job #2764605)
#include <fstream>
#include <vector>
using namespace std;
int n;
int up[100001];
bool fr[100001];
int radacina;
int rez[100001];
vector<int> a[100001];
void read() {
int i, x, y;
ifstream f("cerere.in");
f >> n;
for (i = 1; i <= n; i++)
f >> up[i];
for (i = 1; i < n; i++) {
f >> x >> y;
fr[y] = 1;
a[x].push_back(y);
a[y].push_back(x);
}
f.close();
}
int st[100001], l;
void dfs(int x, int px) {
rez[x] = 1 + rez[st[l - up[x]]];
for (int y : a[x]) {
if (y == px) continue;
st[++l] = y;
dfs(y, x);
l--;
}
}
void solve() {
for (int i = 1; i <= n; i++)
if (fr[i] == 0) {
radacina = i;
break;
}
st[++l] = radacina;
dfs(radacina, 0);
}
void output() {
int i;
ofstream g("cerere.out");
for (i = 1; i <= n; i++)
g << rez[i] - 1 << ' ';
g.close();
}
int main() {
read();
solve();
output();
return 0;
}