結果

問題 No.900 aδδitivee
ユーザー pekempeypekempey
提出日時 2019-10-04 22:19:16
言語 C++14
(gcc 10.1.0 + boost 1.73.0)
結果
AC  
実行時間 126 ms / 2,000 ms
コード長 3,329 Byte
コンパイル時間 1,602 ms
使用メモリ 17,968 KB
最終ジャッジ日時 2020-08-09 06:02:37
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
testcase_00 AC 2 ms
1,604 KB
testcase_01 AC 2 ms
1,604 KB
testcase_02 AC 2 ms
1,620 KB
testcase_03 AC 3 ms
1,628 KB
testcase_04 AC 2 ms
1,624 KB
testcase_05 AC 2 ms
1,620 KB
testcase_06 AC 2 ms
1,628 KB
testcase_07 AC 124 ms
11,836 KB
testcase_08 AC 120 ms
11,832 KB
testcase_09 AC 119 ms
11,832 KB
testcase_10 AC 115 ms
11,836 KB
testcase_11 AC 120 ms
11,832 KB
testcase_12 AC 119 ms
11,832 KB
testcase_13 AC 122 ms
11,836 KB
testcase_14 AC 118 ms
11,836 KB
testcase_15 AC 120 ms
11,832 KB
testcase_16 AC 121 ms
11,836 KB
testcase_17 AC 126 ms
11,832 KB
testcase_18 AC 120 ms
11,832 KB
testcase_19 AC 123 ms
11,836 KB
testcase_20 AC 125 ms
11,832 KB
testcase_21 AC 126 ms
11,832 KB
testcase_22 AC 115 ms
17,964 KB
testcase_23 AC 120 ms
17,964 KB
testcase_24 AC 112 ms
17,960 KB
testcase_25 AC 111 ms
17,960 KB
testcase_26 AC 113 ms
17,968 KB
testcase_27 AC 114 ms
17,964 KB
testcase_28 AC 111 ms
17,968 KB
権限があれば一括ダウンロードができます

ソースコード

diff #
#include <bits/stdc++.h>

#define rep(i, n) for (int i = 0; i < (n); i++)
#define repr(i, n) for (int i = (n) - 1; i >= 0; i--)
#define rep2(i, l, r) for (int i = (l); i < (r); i++)
#define rep2r(i, l, r) for (int i = (r) - 1; i >= (l); i--)
#define range(a) a.begin(), a.end()

using namespace std;
using ll = long long;

struct HLD {
  vector<int> label, parent, head;
  vector<int> L, R;

  HLD(const vector<vector<int>> &g) : label(g.size()), parent(g.size()), head(g.size()), L(g.size()), R(g.size()) {
    const int n = g.size();
    vector<int> size(n, 1);
    auto dfs = [&](auto dfs, int u, int p) -> void {
      for (int v : g[u]) if (v != p) {
        dfs(dfs, v, u);
        size[u] += size[v];
      }
    };
    dfs(dfs, 0, -1);
    int k = 0;
    auto dfs2 = [&](auto dfs, int u, int p, int h) -> void {
      L[u] = k;
      label[u] = k++;
      head[u] = h;
      parent[u] = p;
      for (int v : g[u]) if (v != p && size[v] * 2 >  size[u]) dfs(dfs, v, u, h);
      for (int v : g[u]) if (v != p && size[v] * 2 <= size[u]) dfs(dfs, v, u, v);
      R[u] = k;
    };
    dfs2(dfs2, 0, -1, 0);
  }

  int lca(int u, int v) {
    for (;;) {
      if (label[u] > label[v]) swap(u, v);
      if (head[u] == head[v]) return u;
      v = parent[head[v]];
    }
  }

  template<class F> void each(int u, int v, F f) {
    for (;;) {
      if (label[u] > label[v]) swap(u, v);
      if (head[u] == head[v]) {
        f(label[u], label[v]);
        return;
      }
      f(label[head[v]], label[v]);
      v = parent[head[v]];
    }
  }

  template<class F> void each_edge(int u, int v, F f) {
    for (;;) {
      if (label[u] > label[v]) swap(u, v);
      if (head[u] == head[v]) {
        if (u != v) f(label[u] + 1, label[v]);
        return;
      }
      f(label[head[v]], label[v]);
      v = parent[head[v]];
    }
  }

  int operator[](int u) {
    return label[u];
  };
};

struct bitrange {
  vector<long long> S0;
  vector<long long> S1;

  bitrange(int n) : S0(n + 1), S1(n + 1) {}

  void add0(int k, long long v) {
    for (int i = k + 1; i < S0.size(); i += i & -i) {
      S0[i] -= k * v;
      S1[i] += v;
    }
  }

  long long sum0(int k) {
    long long s0 = 0;
    long long s1 = 0;
    for (int i = k; i > 0; i -= i & -i) {
      s0 += S0[i];
      s1 += S1[i];
    }
    return s0 + s1 * k;
  }

  void add(int l, int r, long long v) {
    add0(l, v);
    add0(r, -v);
  }

  long long sum(int l, int r) {
    return sum0(r) - sum0(l);
  }
};



int main() {
  cin.tie(nullptr);
  ios::sync_with_stdio(false);
  cout << fixed << setprecision(15);
  int N; cin >> N;
  vector<vector<int>> G(N);
  vector<int> A(N - 1), B(N - 1), C(N - 1);
  rep(i, N - 1) {
    cin >> A[i] >> B[i] >> C[i];
    G[A[i]].emplace_back(B[i]);
    G[B[i]].emplace_back(A[i]);
  }
  HLD hld(G);
  bitrange bit(N);
  rep(i, N - 1) {
    if (hld[A[i]] < hld[B[i]]) {
      bit.add(hld[B[i]], hld[B[i]] + 1, C[i]);
    } else {
      bit.add(hld[A[i]], hld[A[i]] + 1, C[i]);
    }
  }
  int Q; cin >> Q;
  while (Q--) {
    int type; cin >> type;
    if (type == 1) {
      int a, x; cin >> a >> x;
      bit.add(hld.L[a] + 1, hld.R[a], x);
    } else {
      int a; cin >> a;
      ll ans = 0;
      hld.each_edge(a, 0, [&](int l, int r) {
        ans += bit.sum(l, r + 1);
      });
      cout << ans << '\n';
    }
  }
}
0