Submission #3334869

---

Source Code Expand

#include <bits/stdc++.h>
using namespace std;
using lint = long long int;
template<class T = int> using V = vector<T>;
template<class T = int> using VV = V< V<T> >;
template<class T> void assign(V<T>& v, int n, const T& a = T()) { v.assign(n, a); }
template<class T, class... U> void assign(V<T>& v, int n, const U&... u) { v.resize(n); for (auto&& i : v) assign(i, u...); }

struct QU {
  V<> par, rank, _size;
  V<lint> res;

  QU(int n) : par(n), rank(n), _size(n, 1), res(n) { 
    iota(par.begin(), par.end(), 0);
  }

  int find(int a) {
    if (par[a] == a) return a;
    return find(par[a]);
  }

  bool same(int a, int b) { return find(a) == find(b); }

  int size(int a) { return _size[find(a)]; }

  void unite(int a, int b, lint c) {
    a = find(a), b = find(b);
    if (a == b) return;
    if (rank[a] < rank[b]) {
      res[b] += _size[a] * c;
      res[a] += _size[b] * c - res[b];
      par[a] = b;
      _size[b] += _size[a];
    } else {
      res[a] += _size[b] * c;
      res[b] += _size[a] * c - res[a];
      par[b] = a;
      _size[a] += _size[b];
    }
    if (rank[a] == rank[b]) rank[a]++;
  }
};

int main() {
  cin.tie(NULL); ios::sync_with_stdio(false);
  int n; cin >> n;
  struct edge { int from, to; lint c; };
  V<edge> es(n - 1);
  for (int i = 0; i < n - 1; i++) {
    int a, b, c; cin >> a >> b >> c, a--, b--;
    es[i] = {a, b, c};
  }
  sort(es.begin(), es.end(), [](edge x, edge y) { return x.c > y.c; });
  QU qu(n);
  for (auto&& e : es) qu.unite(e.from, e.to, e.c);
  VV<> ch(n);
  int root = qu.find(0);
  for (int v = 0; v < n; v++) if (v != root) ch[qu.par[v]].push_back(v);
  queue<int> que;
  que.push(root);
  while (!que.empty()) {
    int v = que.front(); que.pop();
    for (int w : ch[v]) {
      qu.res[w] += qu.res[v]; que.push(w);
    }
  }
  for (int i = 0; i < n; i++) cout << qu.res[i] << '\n';
}

Submission Info

Judge Result