Submission #1837082

---

Source Code Expand

#include <iostream>
#include <algorithm>
#include <vector>
#include <tuple>

template<class T>
struct ICAdd {
  std::vector<T> val;
  std::vector<int> l;
  std::vector<int> r;
  std::vector<int> uf;
  std::vector<int> uf2;

  ICAdd(int n) : l(n, -1), r(n, -1), val(n), uf(n, -1), uf2(n) {
    for (int i = 0; i < n; i++) {
      uf2[i] = i;
    }
  }

  int find(int u) {
    if (uf[u] < 0) return u;
    return uf[u] = find(uf[u]);
  }

  void merge(int u, int v) {
    u = find(u);
    v = find(v);
    if (u == v) return;
    if (-uf[u] < -uf[v]) std::swap(u, v);
    uf[u] += uf[v];
    uf[v] = u;
    val.push_back(T());
    l.push_back(uf2[u]);
    r.push_back(uf2[v]);
    uf2[u] = l.size() - 1;
  }

  void add(int u, T x) {
    val[uf2[find(u)]] += x;
  }

  int size(int u) {
    return -uf[find(u)];
  }

  void dfs(int u, T x) {
    if (u == -1) return;
    val[u] += x;
    dfs(l[u], val[u]);
    dfs(r[u], val[u]);
  }

  void build() {
    std::vector<bool> used(l.size());
    for (int i = l.size() - 1; i >= 0; i--) {
      if (l[i] != -1) used[l[i]] = true;
      if (r[i] != -1) used[r[i]] = true;
      if (!used[i]) {
        dfs(i, T());
      }
    }
  }
};

int main() {
  using namespace std;

  cin.tie(0);
  ios::sync_with_stdio(false);

  int n;
  cin >> n;
 
  vector<tuple<int, int, int>> es;
  for (int i = 0; i < n - 1; i++) {
    int u, v, w;
    cin >> u >> v >> w;
    u--;
    v--;
    es.emplace_back(w, u, v);
  }
 
  sort(es.begin(), es.end());
  reverse(es.begin(), es.end());
 
  ICAdd<long long> ic(n);
  for (auto e : es) {
    int w, u, v;
    tie(w, u, v) = e;
    long long x = ic.size(u);
    long long y = ic.size(v);
    ic.add(u, y*w);
    ic.add(v, x*w);
    ic.merge(u, v);
  }
 
  ic.build();
 
  for (int i = 0; i < n; i++) {
    cout << ic.val[i] << endl;
  }
}

Submission Info

Judge Result