Pagini recente » Cod sursa (job #1022478) | Cod sursa (job #66630) | Cod sursa (job #2731127) | Cod sursa (job #941210) | Cod sursa (job #3148427)
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
const int NMAX = 100003;
vector<int> g[NMAX];
int tata[NMAX], k[NMAX], s[NMAX];
bool vis[NMAX];
int radacina = 1;
void dfs(int nod) {
vis[nod] = true;
if (k[nod] != 0) {
int stramos = nod;
for (int i = 1; i <= k[nod]; i++)
stramos = tata[stramos];
s[nod] = s[stramos] + 1;
}
for (auto fiu : g[nod]) {
if (!vis[fiu])
dfs(fiu);
}
}
int main() {
int n;
fin >> n;
for (int i = 1; i <= n; i++)
fin >> k[i];
for (int i = 1; i <= n - 1; i++) {
int x, y;
fin >> x >> y;
tata[y] = x;
g[x].push_back(y);
}
while (tata[radacina] != 0) {
radacina = tata[radacina];
}
dfs(radacina);
for (int i = 1; i <= n; i++) {
fout << s[i] << ' ';
}
return 0;
}