#include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include //#define int ll #define INT128_MAX (__int128)(((unsigned __int128) 1 << ((sizeof(__int128) * __CHAR_BIT__) - 1)) - 1) #define INT128_MIN (-INT128_MAX - 1) #define pb push_back #define eb emplace_back #define clock chrono::steady_clock::now().time_since_epoch().count() using namespace std; template ostream& print_tuple(ostream& os, const tuple tu) { os << get(tu); if constexpr (I + 1 != sizeof...(args)) { os << ' '; print_tuple(os, tu); } return os; } template ostream& operator<<(ostream& os, const tuple tu) { return print_tuple(os, tu); } template ostream& operator<<(ostream& os, const pair pr) { return os << pr.first << ' ' << pr.second; } template ostream& operator<<(ostream& os, const array &arr) { for(size_t i = 0; T x : arr) { os << x; if (++i != N) os << ' '; } return os; } template ostream& operator<<(ostream& os, const vector &vec) { for(size_t i = 0; T x : vec) { os << x; if (++i != size(vec)) os << ' '; } return os; } template ostream& operator<<(ostream& os, const set &s) { for(size_t i = 0; T x : s) { os << x; if (++i != size(s)) os << ' '; } return os; } template ostream& operator<<(ostream& os, const multiset &s) { for(size_t i = 0; T x : s) { os << x; if (++i != size(s)) os << ' '; } return os; } template ostream& operator<<(ostream& os, const map &m) { for(size_t i = 0; pair x : m) { os << x.first << " : " << x.second; if (++i != size(m)) os << ", "; } return os; } template ostream& operator<<(ostream&os, span &s) { for(size_t i = 0; T &x : s) { os << x; if (++i != size(s)) os << ' '; } return os; } #ifdef DEBUG #define dbg(...) cerr << '(', _do(#__VA_ARGS__), cerr << ") = ", _do2(__VA_ARGS__) template void _do(T &&x) { cerr << x; } template void _do(T &&x, S&&...y) { cerr << x << ", "; _do(y...); } template void _do2(T &&x) { cerr << x << endl; } template void _do2(T &&x, S&&...y) { cerr << x << ", "; _do2(y...); } #else #define dbg(...) #endif using ll = long long; using ull = unsigned long long; using ldb = long double; using pii = pair; using pll = pair; //#define double ldb template using vc = vector; template using vvc = vc>; template using vvvc = vc>; using vi = vc; using vll = vc; using vvi = vvc; using vvll = vvc; template using min_heap = priority_queue, greater>; template using max_heap = priority_queue; template concept R_invocable = requires(F&& f, Args&&... args) { { std::invoke(std::forward(f), std::forward(args)...) } -> std::same_as; }; template, typename F> requires R_invocable void pSum(rng &&v, F f) { if (!v.empty()) for(T p = *v.begin(); T &x : v | views::drop(1)) x = p = f(p, x); } template> void pSum(rng &&v) { if (!v.empty()) for(T p = *v.begin(); T &x : v | views::drop(1)) x = p = p + x; } template void Unique(rng &v) { ranges::sort(v); v.resize(unique(v.begin(), v.end()) - v.begin()); } template rng invPerm(rng p) { rng ret = p; for(int i = 0; i < ssize(p); i++) ret[p[i]] = i; return ret; } template vi argSort(rng p) { vi id(size(p)); iota(id.begin(), id.end(), 0); ranges::sort(id, {}, [&](int i) { return pair(p[i], i); }); return id; } template, typename F> requires invocable vi argSort(rng p, F proj) { vi id(size(p)); iota(id.begin(), id.end(), 0); ranges::sort(id, {}, [&](int i) { return pair(proj(p[i]), i); }); return id; } template vvi read_graph(int n, int m, int base) { vvi g(n); for(int i = 0; i < m; i++) { int u, v; cin >> u >> v; u -= base, v -= base; g[u].emplace_back(v); if constexpr (!directed) g[v].emplace_back(u); } return g; } template vvi adjacency_list(int n, vc e, int base) { vvi g(n); for(auto [u, v] : e) { u -= base, v -= base; g[u].emplace_back(v); if constexpr (!directed) g[v].emplace_back(u); } return g; } template vc equal_subarrays(vc &v) { vc lr; for(int i = 0, j = 0; i < ssize(v); i = j) { while(j < ssize(v) and v[i] == v[j]) j++; lr.eb(i, j); } return lr; } template requires invocable vc equal_subarrays(vc &v, F proj) { vc lr; for(int i = 0, j = 0; i < ssize(v); i = j) { while(j < ssize(v) and proj(v[i]) == proj(v[j])) j++; lr.eb(i, j); } return lr; } template void setBit(T &msk, int bit, bool x) { (msk &= ~(T(1) << bit)) |= T(x) << bit; } template void onBit(T &msk, int bit) { setBit(msk, bit, true); } template void offBit(T &msk, int bit) { setBit(msk, bit, false); } template void flipBit(T &msk, int bit) { msk ^= T(1) << bit; } template bool getBit(T msk, int bit) { return msk >> bit & T(1); } template T floorDiv(T a, T b) { if (b < 0) a *= -1, b *= -1; return a >= 0 ? a / b : (a - b + 1) / b; } template T ceilDiv(T a, T b) { if (b < 0) a *= -1, b *= -1; return a >= 0 ? (a + b - 1) / b : a / b; } template bool chmin(T &a, T b) { return a > b ? a = b, 1 : 0; } template bool chmax(T &a, T b) { return a < b ? a = b, 1 : 0; } //constantly used templates struct HLD { int n, root; vi dep, sz, p, head, tin, tout, inv_tin, child_list, c; vc lb; inline int head_parent(int v) const { return p[head[v]]; } HLD(vc e, int _root = 0) : root(_root) { precompute(e); } HLD(vi _p) { vc e; root = -1; for(int v = 0; v < ssize(_p); v++) { if (_p[v] == -1 or _p[v] == v) root = v; else e.eb(v, _p[v]); } assert(root != -1); precompute(e); } void precompute(vc &e) { n = ssize(e) + 1; dep = p = head = tin = tout = vi(n); sz = vi(n, 1); vi mx_child_sz(n, -1); { vi d(n); for(auto [u, v] : e) p[u] ^= v, p[v] ^= u, d[u]++, d[v]++; d[root] = 0; for(int i = 0; i < n; i++) { int v = i; while(d[v] == 1) { d[v] = 0, d[p[v]]--, p[p[v]] ^= v; sz[p[v]] += sz[v]; chmax(mx_child_sz[p[v]], sz[v]); v = p[v]; } } p[root] = root; } vi ord(n); { vi f(n + 2); for(int x : sz) f[x + 1]++; pSum(f); for(int v = 0; v < n; v++) ord[n - 1 - (f[sz[v]]++)] = v; } { head[root] = root, tout[root] = n; vi add(n, 1); for(int v : ord | views::drop(1)) { dep[v] = dep[p[v]] + 1; tin[v] = tin[p[v]] + add[p[v]]; add[p[v]] += sz[v]; tout[v] = tin[v] + sz[v]; if (mx_child_sz[p[v]] == sz[v]) mx_child_sz[p[v]] = 0, head[v] = head[p[v]]; else head[v] = v; } } inv_tin = invPerm(tin); lb = vc(n + 1); child_list = vi(n + 1); for(int v = 0; v < n; v++) if (v != root) lb[p[v]]++; pSum(lb); for(int v = 0; v < n; v++) if (v != root and head[v] == v) child_list[--lb[p[v]]] = v; for(int v = 0; v < n; v++) if (v != root and head[v] != v) child_list[--lb[p[v]]] = v; } auto query_path(int u, int v, bool edge = false) { vc lr; while(head[u] != head[v]) { if (dep[head[u]] > dep[head[v]]) swap(u, v); lr.emplace_back(tin[head[v]], tin[v] + 1); v = head_parent(v); } if (tin[u] > tin[v]) swap(u, v); if (tin[u] + edge <= tin[v]) lr.emplace_back(tin[u] + edge, tin[v] + 1); return lr; } //l < r: op(l, op(l + 1, ...)) //l > r: op(r - 1, op(r - 2, ...)) auto query_path_non_commutative(int u, int v, bool edge = false) { vc lr1, lr2; while(head[u] != head[v]) { if (dep[head[u]] > dep[head[v]]) { lr1.emplace_back(tin[u] + 1, tin[head[u]]); u = head_parent(u); } else { lr2.emplace_back(tin[head[v]], tin[v] + 1); v = head_parent(v); } } if (tin[u] + edge <= tin[v]) lr2.emplace_back(tin[u] + edge, tin[v] + 1); else if (tin[v] + edge <= tin[u]) lr1.emplace_back(tin[u] + 1, tin[v] + edge); lr1.insert(end(lr1), lr2.rbegin(), lr2.rend()); return lr1; } auto query_subtree(int v) { return pii(tin[v], tout[v]); } int query_point(int v) { return tin[v]; } int lca(int u, int v) { while(head[u] != head[v]) { if (dep[head[u]] > dep[head[v]]) swap(u, v); v = head_parent(v); } return tin[u] < tin[v] ? u : v; } int dis(int u, int v) { return dep[u] + dep[v] - 2 * dep[lca(u, v)]; } int kth(int s, int t, int k) { int l = lca(s, t); if (int d = dep[s] + dep[t] - 2 * dep[l]; k > d) return -1; else if (k > dep[s] - dep[l]) k = d - k, swap(s, t); while(k > dep[s] - dep[head[s]]) { k -= dep[s] - dep[head[s]] + 1; s = head_parent(s); } return inv_tin[tin[s] - k]; } int median(int u, int v, int w) { return lca(u, v) ^ lca(u, w) ^ lca(v, w); } template vc reorder_init(vc init) { assert(ssize(init) == ssize(dep)); auto r = init; for(int i = 0; i < ssize(init); i++) r[tin[i]] = init[i]; return r; } const span childs(int v) { return span(child_list.begin() + lb[v], lb[v + 1] - lb[v]); } const span light_childs(int v) { return span(child_list.begin() + lb[v] + 1, max(lb[v + 1] - lb[v] - 1, 0)); } inline int heavy_child(int v) { return lb[v] == lb[v + 1] ? -1 : child_list[lb[v]]; } inline int parent(int v) { return p[v]; } inline int depth(int v) { return dep[v]; } inline int size(int v) { return sz[v]; } bool in_subtree_of(int a, int b) { return tin[b] <= tin[a] and tout[a] <= tout[b]; } const span centroid() { if (c.empty()) { vc ok(n, true); for(int v = 0; v < n; v++) { if (2 * (n - sz[v]) > n) ok[v] = false; if (v != root and 2 * sz[v] > n) ok[p[v]] = false; } for(int v = 0; v < n; v++) if (ok[v]) c.eb(v); } return c; } }; template struct MontgomeryModInt { using mint = MontgomeryModInt; using i32 = int32_t; using u32 = uint32_t; using u64 = uint64_t; static constexpr u32 get_r() { u32 res = 1, base = mod; for(i32 i = 0; i < 31; i++) res *= base, base *= base; return -res; } static constexpr u32 get_mod() { return mod; } static constexpr u32 n2 = -u64(mod) % mod; //2^64 % mod static constexpr u32 r = get_r(); //-P^{-1} % 2^32 u32 a; static u32 reduce(const u64 &b) { return (b + u64(u32(b) * r) * mod) >> 32; } static u32 transform(const u64 &b) { return reduce(u64(b) * n2); } MontgomeryModInt() : a(0) {} MontgomeryModInt(const int64_t &b) : a(transform(b % mod + mod)) {} mint pow(u64 k) const { mint res(1), base(*this); while(k) { if (k & 1) res *= base; base *= base, k >>= 1; } return res; } mint inverse() const { return (*this).pow(mod - 2); } u32 get() const { u32 res = reduce(a); return res >= mod ? res - mod : res; } mint& operator+=(const mint &b) { if (i32(a += b.a - 2 * mod) < 0) a += 2 * mod; return *this; } mint& operator-=(const mint &b) { if (i32(a -= b.a) < 0) a += 2 * mod; return *this; } mint& operator*=(const mint &b) { a = reduce(u64(a) * b.a); return *this; } mint& operator/=(const mint &b) { a = reduce(u64(a) * b.inverse().a); return *this; } mint operator-() { return mint() - mint(*this); } bool operator==(mint b) const { return (a >= mod ? a - mod : a) == (b.a >= mod ? b.a - mod : b.a); } bool operator!=(mint b) const { return (a >= mod ? a - mod : a) != (b.a >= mod ? b.a - mod : b.a); } friend mint operator+(mint c, mint d) { return c += d; } friend mint operator-(mint c, mint d) { return c -= d; } friend mint operator*(mint c, mint d) { return c *= d; } friend mint operator/(mint c, mint d) { return c /= d; } friend ostream& operator<<(ostream& os, const mint& b) { return os << b.get(); } friend istream& operator>>(istream& is, mint& b) { int64_t val; is >> val; b = mint(val); return is; } }; //using mint = MontgomeryModInt<998'244'353>; template struct dynamic_kth { multiset D, U; T Dsum = 0, Usum = 0; void slide(int k) { //make |D| = k while(ssize(D) > k) { T x = *D.rbegin(); Usum += x, Dsum -= x; U.insert(x); D.erase(prev(D.end())); } while(ssize(D) < k and !U.empty()) { T x = *U.begin(); Dsum += x, Usum -= x; D.insert(x); U.erase(U.begin()); } } void insert(T x) { if ((U.empty() and (D.empty() or x > *D.rbegin())) or x >= *U.begin()) U.insert(x), Usum += x; else D.insert(x), Dsum += x; } void erase(T x) { if (auto ite = U.find(x); ite != U.end()) U.erase(ite), Usum -= x; else if (auto ite2 = D.find(x); ite2 != D.end()) D.erase(ite2), Dsum -= x; } T kth(int k) { slide(k); return *U.begin(); } T min() { return D.empty() ? *U.begin() : *D.begin(); } T max() { return U.empty() ? *D.rbegin() : *U.rbegin(); } uint64_t size() { return ssize(D) + ssize(U); } }; signed main() { ios::sync_with_stdio(false), cin.tie(NULL); int n; cin >> n; vector a(n + 1); for(int &x : a) cin >> x; vc e(n); for(auto &[u, v] : e) { cin >> u >> v; } HLD hld(e); dynamic_kth s; vector ans(n + 1, LLONG_MAX); auto dfs = [&](int v, auto self) -> void { s.insert(a[v]); if (s.size() > 2 and s.min() != s.max()) { int mn = s.min(); s.erase(mn); int med = s.kth((ssize(s.U) + ssize(s.D)) / 2); if (med == mn) med++; ans[v] = min(ans[v], (ssize(s.D) - ssize(s.U)) * med - s.Dsum + s.Usum); s.insert(mn); } if (s.size() > 2 and s.min() != s.max()) { int mx = s.max(); s.erase(mx); int med = s.kth((ssize(s.U) + ssize(s.D)) / 2); if (med == mx) med--; ans[v] = min(ans[v], (ssize(s.D) - ssize(s.U)) * med - s.Dsum + s.Usum); s.insert(mx); } if (s.size() > 2 and s.min() == s.max()) ans[v] = 1; for(int x : hld.childs(v)) self(x, self); s.erase(a[v]); }; dfs(0, dfs); for(int i = 1; i <= n; i++) cout << (ans[i] == LLONG_MAX ? -1 : ans[i]) << '\n'; return 0; }