#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 using namespace std; using lint = long long; using pint = pair; using plint = pair; struct fast_ios { fast_ios(){ cin.tie(nullptr), ios::sync_with_stdio(false), cout << fixed << setprecision(20); }; } fast_ios_; #define ALL(x) (x).begin(), (x).end() #define FOR(i, begin, end) for(int i=(begin),i##_end_=(end);i=i##_begin_;i--) #define REP(i, n) FOR(i,0,n) #define IREP(i, n) IFOR(i,0,n) template bool chmax(T &m, const T q) { return m < q ? (m = q, true) : false; } template bool chmin(T &m, const T q) { return m > q ? (m = q, true) : false; } const std::vector> grid_dxs{{1, 0}, {-1, 0}, {0, 1}, {0, -1}}; int floor_lg(long long x) { return x <= 0 ? -1 : 63 - __builtin_clzll(x); } template T1 floor_div(T1 num, T2 den) { return (num > 0 ? num / den : -((-num + den - 1) / den)); } template std::pair operator+(const std::pair &l, const std::pair &r) { return std::make_pair(l.first + r.first, l.second + r.second); } template std::pair operator-(const std::pair &l, const std::pair &r) { return std::make_pair(l.first - r.first, l.second - r.second); } template std::vector sort_unique(std::vector vec) { sort(vec.begin(), vec.end()), vec.erase(unique(vec.begin(), vec.end()), vec.end()); return vec; } template int arglb(const std::vector &v, const T &x) { return std::distance(v.begin(), std::lower_bound(v.begin(), v.end(), x)); } template int argub(const std::vector &v, const T &x) { return std::distance(v.begin(), std::upper_bound(v.begin(), v.end(), x)); } template IStream &operator>>(IStream &is, std::vector &vec) { for (auto &v : vec) is >> v; return is; } template OStream &operator<<(OStream &os, const std::vector &vec); template OStream &operator<<(OStream &os, const std::array &arr); template OStream &operator<<(OStream &os, const std::unordered_set &vec); template OStream &operator<<(OStream &os, const pair &pa); template OStream &operator<<(OStream &os, const std::deque &vec); template OStream &operator<<(OStream &os, const std::set &vec); template OStream &operator<<(OStream &os, const std::multiset &vec); template OStream &operator<<(OStream &os, const std::unordered_multiset &vec); template OStream &operator<<(OStream &os, const std::pair &pa); template OStream &operator<<(OStream &os, const std::map &mp); template OStream &operator<<(OStream &os, const std::unordered_map &mp); template OStream &operator<<(OStream &os, const std::tuple &tpl); template OStream &operator<<(OStream &os, const std::vector &vec) { os << '['; for (auto v : vec) os << v << ','; os << ']'; return os; } template OStream &operator<<(OStream &os, const std::array &arr) { os << '['; for (auto v : arr) os << v << ','; os << ']'; return os; } template std::istream &operator>>(std::istream &is, std::tuple &tpl) { std::apply([&is](auto &&... args) { ((is >> args), ...);}, tpl); return is; } template OStream &operator<<(OStream &os, const std::tuple &tpl) { os << '('; std::apply([&os](auto &&... args) { ((os << args << ','), ...);}, tpl); return os << ')'; } template OStream &operator<<(OStream &os, const std::unordered_set &vec) { os << '{'; for (auto v : vec) os << v << ','; os << '}'; return os; } template OStream &operator<<(OStream &os, const std::deque &vec) { os << "deq["; for (auto v : vec) os << v << ','; os << ']'; return os; } template OStream &operator<<(OStream &os, const std::set &vec) { os << '{'; for (auto v : vec) os << v << ','; os << '}'; return os; } template OStream &operator<<(OStream &os, const std::multiset &vec) { os << '{'; for (auto v : vec) os << v << ','; os << '}'; return os; } template OStream &operator<<(OStream &os, const std::unordered_multiset &vec) { os << '{'; for (auto v : vec) os << v << ','; os << '}'; return os; } template OStream &operator<<(OStream &os, const std::pair &pa) { return os << '(' << pa.first << ',' << pa.second << ')'; } template OStream &operator<<(OStream &os, const std::map &mp) { os << '{'; for (auto v : mp) os << v.first << "=>" << v.second << ','; os << '}'; return os; } template OStream &operator<<(OStream &os, const std::unordered_map &mp) { os << '{'; for (auto v : mp) os << v.first << "=>" << v.second << ','; os << '}'; return os; } #ifdef HITONANODE_LOCAL const string COLOR_RESET = "\033[0m", BRIGHT_GREEN = "\033[1;32m", BRIGHT_RED = "\033[1;31m", BRIGHT_CYAN = "\033[1;36m", NORMAL_CROSSED = "\033[0;9;37m", RED_BACKGROUND = "\033[1;41m", NORMAL_FAINT = "\033[0;2m"; #define dbg(x) std::cerr << BRIGHT_CYAN << #x << COLOR_RESET << " = " << (x) << NORMAL_FAINT << " (L" << __LINE__ << ") " << __FILE__ << COLOR_RESET << std::endl #define dbgif(cond, x) ((cond) ? std::cerr << BRIGHT_CYAN << #x << COLOR_RESET << " = " << (x) << NORMAL_FAINT << " (L" << __LINE__ << ") " << __FILE__ << COLOR_RESET << std::endl : std::cerr) #else #define dbg(x) ((void)0) #define dbgif(cond, x) ((void)0) #endif struct Manager { lint lo_sum = 0; lint hi_sum = 0; multiset lo; multiset hi; lint solve() { if (size() <= 1) return 0; const int nlo = lo.size(); const int nhi = hi.size(); lint med = *lo.rbegin(); lint ret = hi_sum - med * nhi + med * nlo - lo_sum; if (hi.size()) { med = *hi.begin(); chmin(ret, hi_sum - med * nhi + med * nlo - lo_sum); } return ret; } void balance() { while (!lo.empty() and !hi.empty() and *lo.rbegin() > *hi.begin()) { int max_lo = *lo.rbegin(); int min_hi = *hi.begin(); lo.erase(prev(lo.end())); hi.erase(hi.begin()); lo.insert(min_hi); hi.insert(max_lo); lo_sum += min_hi - max_lo; hi_sum += max_lo - min_hi; } while (lo.size() > hi.size() + 1) { hi.insert(*lo.rbegin()); hi_sum += *lo.rbegin(); lo_sum -= *lo.rbegin(); lo.erase(prev(lo.end())); } while (hi.size() > lo.size()) { lo.insert(*hi.begin()); lo_sum += *hi.begin(); hi_sum -= *hi.begin(); hi.erase(hi.begin()); } } void add(int x) { if (lo.empty() or x <= *lo.rbegin()) { lo.insert(x); lo_sum += x; } else { hi.insert(x); hi_sum += x; } balance(); } void erase(int x) { if (lo.count(x)) { lo.erase(lo.find(x)); lo_sum -= x; } else { hi.erase(hi.find(x)); hi_sum -= x; } balance(); } int size() const { return lo.size() + hi.size(); } int min() const { return *lo.begin(); } int max() const { return *hi.rbegin(); } }; int main() { int N; cin >> N; vector A(N + 1); for (auto &a : A) cin >> a; vector> to(N + 1); for (int e = 0; e < N; ++e) { int a, b; cin >> a >> b; to.at(a).push_back(b); to.at(b).push_back(a); } Manager manager; vector ret(N, -1); auto rec = [&](auto &&self, int now, int prv) -> void { manager.add(A.at(now)); if (now) { if (manager.size() <= 2) { ret.at(now - 1) = -1; } else { dbg(make_tuple(now, manager.lo, manager.hi, manager.lo_sum, manager.hi_sum)); int max = manager.max(); int min = manager.min(); if (max == min) { ret.at(now - 1) = 1; } else { manager.erase(max); lint tmp = manager.solve(); manager.add(max); manager.erase(min); chmin(tmp, manager.solve()); manager.add(min); ret.at(now - 1) = tmp; } } } for (int nxt : to.at(now)) { if (nxt == prv) continue; self(self, nxt, now); } manager.erase(A.at(now)); }; rec(rec, 0, -1); for (auto r : ret) cout << r << '\n'; }