Pagini recente » Cod sursa (job #2017565) | Cod sursa (job #1079874) | Cod sursa (job #1892212) | Cod sursa (job #849857) | Cod sursa (job #2208974)
#include <bits/stdc++.h>
#define all(cont) cont.begin(), cont.end()
#define pb push_back
#define popb pop_back
using namespace std;
ifstream f ("cerere.in");
ofstream g ("cerere.out");
const int NMAX = 1e5;
int dp[NMAX], v[NMAX];
bool child[NMAX];
vector <int> adj[NMAX];
vector <int> path;
void dfs (int node) {
path.pb(node);
for (auto &i : adj[node]) {
if (v[i] != 0)
dp[i] = 1 + dp[path[path.size() - v[i]]];
else
dp[i] = 0;
dfs(i);
}
path.popb();
}
int main() {
int n, start;
f >> n;
memset (dp, -1, sizeof(dp));
for (int i = 0; i < n; ++i) {
f >> v[i];
if (v[i] == 0) {
dp[i] = 0;
}
}
for (int i = 0; i < n - 1; ++i) {
int u, v;
f >> u >> v;
--u; --v;
adj[u].pb(v);
child[v] = true;
}
for (int i = 0; i < n; ++i) {
if (!child[i]) {
start = i;
}
}
dfs(start);
for (int i = 0; i < n; ++i) {
g << dp[i] << ' ';
}
f.close();
g.close();
return 0;
}