Pagini recente » Cod sursa (job #713980) | Cod sursa (job #1991155) | Cod sursa (job #1358562) | Cod sursa (job #1066626) | Cod sursa (job #1770735)
#include <fstream>
#include <vector>
using namespace std;
class InputReader {
public:
InputReader() {}
InputReader(const char *file_name) {
input_file = fopen(file_name, "r");
cursor = 0;
fread(buffer, SIZE, 1, input_file);
}
inline InputReader &operator >>(int &n) {
while(buffer[cursor] < '0' || buffer[cursor] > '9') {
advance();
}
n = 0;
while('0' <= buffer[cursor] && buffer[cursor] <= '9') {
n = n * 10 + buffer[cursor] - '0';
advance();
}
return *this;
}
private:
FILE *input_file;
static const int SIZE = 1 << 17;
int cursor;
char buffer[SIZE];
inline void advance() {
++ cursor;
if(cursor == SIZE) {
cursor = 0;
fread(buffer, SIZE, 1, input_file);
}
}
};
const int NMAX = 100000 + 5;
vector <int> tree[NMAX];
int k[NMAX];
int ans[NMAX];
int stk[NMAX];
int stkSz;
int father[NMAX];
int h[NMAX];
void dfs(int node) {
if (k[node])
h[node] = 1 + h[stk[stkSz - k[node]]];
else
h[node] = 0;
for (auto it: tree[node]) {
stk[++ stkSz] = it;
dfs(it);
-- stkSz;
}
}
int main()
{
InputReader cin("cerere.in");
ofstream cout("cerere.out");
int n;
cin >> n;
for (int i = 1; i <= n; ++ i)
cin >> k[i];
for (int i = 1; i < n; ++ i) {
int a, b;
cin >> a >> b;
father[b] = a;
tree[a].push_back(b);
}
int one = 1;
for (int i = 1; i <= n; ++ i)
if (!father[i])
one = i;
stk[++ stkSz] = one;
dfs(one);
for (int i = 1; i <= n; ++ i)
cout << h[i] << " \n"[i == n];
//cin.close();
cout.close();
return 0;
}