結果

問題 No.1216 灯籠流し/Lanterns
ユーザー risujirohrisujiroh
提出日時 2020-08-30 16:03:18
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 2,865 ms / 4,500 ms
コード長 6,798 bytes
コンパイル時間 8,424 ms
コンパイル使用メモリ 457,624 KB
実行使用メモリ 107,748 KB
最終ジャッジ日時 2023-08-09 14:47:44
合計ジャッジ時間 57,653 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,328 KB
testcase_01 AC 1 ms
4,332 KB
testcase_02 AC 2 ms
4,332 KB
testcase_03 AC 2 ms
4,332 KB
testcase_04 AC 36 ms
17,864 KB
testcase_05 AC 1,587 ms
100,668 KB
testcase_06 AC 81 ms
13,384 KB
testcase_07 AC 1,493 ms
86,292 KB
testcase_08 AC 1,320 ms
68,732 KB
testcase_09 AC 539 ms
44,256 KB
testcase_10 AC 607 ms
43,020 KB
testcase_11 AC 227 ms
31,296 KB
testcase_12 AC 34 ms
15,160 KB
testcase_13 AC 253 ms
39,676 KB
testcase_14 AC 97 ms
22,504 KB
testcase_15 AC 208 ms
29,732 KB
testcase_16 AC 155 ms
33,932 KB
testcase_17 AC 1,123 ms
52,048 KB
testcase_18 AC 220 ms
30,364 KB
testcase_19 AC 1,757 ms
90,240 KB
testcase_20 AC 621 ms
53,408 KB
testcase_21 AC 275 ms
36,592 KB
testcase_22 AC 1,347 ms
64,644 KB
testcase_23 AC 47 ms
22,696 KB
testcase_24 AC 67 ms
12,736 KB
testcase_25 AC 31 ms
9,360 KB
testcase_26 AC 611 ms
40,288 KB
testcase_27 AC 145 ms
26,972 KB
testcase_28 AC 86 ms
24,488 KB
testcase_29 AC 247 ms
39,056 KB
testcase_30 AC 437 ms
46,348 KB
testcase_31 AC 1,005 ms
59,512 KB
testcase_32 AC 385 ms
48,704 KB
testcase_33 AC 77 ms
20,312 KB
testcase_34 AC 1,681 ms
105,584 KB
testcase_35 AC 2,060 ms
106,340 KB
testcase_36 AC 2,865 ms
106,852 KB
testcase_37 AC 600 ms
69,376 KB
testcase_38 AC 1,721 ms
104,680 KB
testcase_39 AC 1,851 ms
104,108 KB
testcase_40 AC 1,890 ms
106,548 KB
testcase_41 AC 1,958 ms
107,124 KB
testcase_42 AC 1,883 ms
106,048 KB
testcase_43 AC 1,898 ms
107,748 KB
testcase_44 AC 1,950 ms
107,432 KB
testcase_45 AC 1,913 ms
106,868 KB
testcase_46 AC 1,938 ms
106,932 KB
testcase_47 AC 1,919 ms
106,268 KB
testcase_48 AC 1,892 ms
105,760 KB
testcase_49 AC 1,907 ms
105,580 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

// #ifdef ONLINE_JUDGE
#pragma GCC optimize("Ofast")
#pragma GCC target("avx2,bmi,bmi2,lzcnt")
// #endif

#include <bits/extc++.h>
#include <x86intrin.h>

#ifndef DUMP
#define DUMP(...) void(0)
#endif

using namespace std;

struct graph {
  struct edge {
    int src, dst;
    int64_t cost;
    int operator-(int v) const { return src ^ dst ^ v; }
  };
  int n, m;
  vector<edge> edges;
  vector<vector<pair<int, int>>> adj;
  graph(int _n = 0) : n(_n), m(0), adj(n) {}
  int add(const edge& e, bool directed = false) {
    edges.push_back(e);
    adj[e.src].emplace_back(m, e.dst);
    if (not directed) adj[e.dst].emplace_back(m, e.src);
    return m++;
  }
};

struct dfs_forest : graph {
  using T = decltype(edge::cost);
  vector<int> root, pv, pe, sz, dep, min_dep, last, ord, in, out;
  vector<T> dist;
  int trials;
  dfs_forest(int _n = 0) : graph(_n), dist(n), trials(0) {
    for (auto p : {&root, &pv, &pe, &sz, &dep, &min_dep, &last, &in, &out})
      p->assign(n, -1);
  }
  int add(const edge& e) { return graph::add(e); }
  void dfs(int v) {
    sz[v] = 1, min_dep[v] = dep[v], last[v] = trials;
    in[v] = size(ord), ord.push_back(v);
    for (auto [id, u] : adj[v]) {
      if (id == pe[v]) continue;
      if (last[u] == trials) {
        min_dep[v] = min(min_dep[v], dep[u]);
        continue;
      }
      root[u] = root[v], pv[u] = v, pe[u] = id, dep[u] = dep[v] + 1;
      dist[u] = dist[v] + edges[id].cost;
      dfs(u);
      sz[v] += sz[u], min_dep[v] = min(min_dep[v], min_dep[u]);
    }
    out[v] = size(ord);
  }
  void build(int r, bool clear_ord = true) {
    root[r] = r, pv[r] = pe[r] = -1, dep[r] = 0, dist[r] = T{};
    if (clear_ord) ord.clear();
    dfs(r);
    ++trials;
  }
  void build() {
    fill(begin(root), end(root), -1);
    for (int v = 0; v < n; ++v)
      if (root[v] == -1) build(v, v == 0);
  }
  int farther(int id) const {
    int u = edges[id].src, v = edges[id].dst;
    return dep[u] < dep[v] ? v : u;
  }
  bool spans(int id) const { return id == pe[farther(id)]; }
  bool anc(int u, int v) const { return in[u] <= in[v] and out[v] <= out[u]; }
};

template <class T>
struct range2d {
  struct bit_vector {
    using size_type = uint32_t;
    static constexpr size_type wsize = 64;
    static size_type rank64(uint64_t x, size_type i) {
      return _popcnt64(_bzhi_u64(x, i));
    }
#pragma pack(4)
    struct block_t {
      uint64_t bit;
      size_type sum;
    };
#pragma pack()
    size_type n, zeros;
    vector<block_t> block;
    bit_vector(size_type _n = 0) : n(_n), block(n / wsize + 1) {}
    int operator[](size_type i) const {
      return block[i / wsize].bit >> i % wsize & 1;
    }
    void set(size_type i) { block[i / wsize].bit |= (uint64_t)1 << i % wsize; }
    void build() {
      for (size_type j = 0; j < n / wsize; ++j)
        block[j + 1].sum = block[j].sum + _popcnt64(block[j].bit);
      zeros = rank0(n);
    }
    size_type rank0(size_type i) const { return i - rank1(i); }
    size_type rank1(size_type i) const {
      auto&& e = block[i / wsize];
      return e.sum + rank64(e.bit, i % wsize);
    }
  };
  struct fenwick {
    int n;
    vector<T> t;
    fenwick(int _n = 0) : n(_n), t(n) {}
    void add(int i, T val) {
      for (++i; i <= n; i += _blsi_u64(i)) t[i - 1] += val;
    }
    T sum(int i) const {
      T res = 0;
      for (; i; i = _blsr_u64(i)) res += t[i - 1];
      return res;
    }
    T sum(int l, int r) const { return sum(r) - sum(l); }
  };
  using key_type = int64_t;
  int n, lg;
  vector<bit_vector> bv;
  vector<fenwick> ft;
  vector<pair<key_type, key_type>> p;
  vector<key_type> ys;
  void add_point(key_type x, key_type y) {
    p.emplace_back(x, y);
    ys.push_back(y);
  }
  void build() {
    sort(begin(p), end(p));
    p.erase(unique(begin(p), end(p)), end(p));
    n = size(p);
    sort(begin(ys), end(ys));
    ys.erase(unique(begin(ys), end(ys)), end(ys));
    vector<int> a(n), na(n);
    for (int i = 0; i < n; ++i) a[i] = y2a(p[i].second);
    lg = __lg(max(*max_element(begin(a), end(a)), 1)) + 1;
    bv.assign(lg, n);
    ft.assign(lg + 1, n);
    for (auto h = lg; h--;) {
      for (int i = 0; i < n; ++i)
        if (a[i] >> h & 1) bv[h].set(i);
      bv[h].build();
      array it{begin(na), begin(na) + bv[h].zeros};
      for (int i = 0; i < n; ++i) *it[bv[h][i]]++ = a[i];
      swap(a, na);
    }
  }
  int x2i(key_type x) const {
    pair t{x, numeric_limits<key_type>::min()};
    return lower_bound(begin(p), end(p), t) - begin(p);
  }
  int y2a(key_type y) const {
    return lower_bound(begin(ys), end(ys), y) - begin(ys);
  }
  void add(key_type x, key_type y, T val) {
    assert(binary_search(begin(p), end(p), pair{x, y}));
    int i = lower_bound(begin(p), end(p), pair{x, y}) - begin(p);
    ft[lg].add(i, val);
    for (int h = lg; h--;) {
      int i0 = bv[h].rank0(i);
      if (bv[h][i] == 0)
        i = i0;
      else
        i += bv[h].zeros - i0;
      ft[h].add(i, val);
    }
  }
  T sum(int l, int r, int ub) const {
    if (ub >= 1 << lg) return ft[lg].sum(l, r);
    T res = 0;
    for (int h = lg; h--;) {
      int l0 = bv[h].rank0(l), r0 = bv[h].rank0(r);
      if (~ub >> h & 1)
        l = l0, r = r0;
      else {
        res += ft[h].sum(l0, r0);
        l += bv[h].zeros - l0;
        r += bv[h].zeros - r0;
      }
    }
    return res;
  }
  T sum(key_type lx, key_type rx, key_type ly, key_type ry) const {
    int l = x2i(lx), r = x2i(rx);
    return sum(l, r, y2a(ry)) - sum(l, r, y2a(ly));
  }
};

int main() {
  cin.tie(nullptr)->sync_with_stdio(false);
  int n, q;
  cin >> n >> q;
  dfs_forest g(n);
  for (int _ = n - 1; _--;) {
    int u, v;
    int64_t w;
    cin >> u >> v >> w;
    --u, --v;
    g.add({u, v, w});
  }
  g.build();

  struct query {
    int type, v;
    int64_t t, l;
  };
  vector<query> queries(q);
  vector<range2d<int>> ds(2 * n);
  for (auto&& [type, v, t, l] : queries) {
    cin >> type >> v >> t >> l;
    --v;
    DUMP(type, v, t, l);
    if (type == 0) {
      int i = g.in[v] + n;
      ds[i].add_point(t + g.dist[v], g.dist[v] - l);
      while (i >>= 1) ds[i].add_point(t + g.dist[v], g.dist[v] - l);
    }
  }
  for (auto&& e : ds)
    if (not empty(e.p)) e.build();

  for (auto&& [type, v, t, l] : queries) {
    if (type == 0) {
      int i = g.in[v] + n;
      ds[i].add(t + g.dist[v], g.dist[v] - l, 1);
      while (i >>= 1) ds[i].add(t + g.dist[v], g.dist[v] - l, 1);
    } else if (type == 1) {
      int res = 0;
      for (int l = g.in[v] + n, r = g.out[v] + n; l < r; l >>= 1, r >>= 1) {
        if (l & 1)
          res += ds[l++].sum(-9e18, t + g.dist[v] + 1, -9e18, g.dist[v] + 1);
        if (r & 1)
          res += ds[--r].sum(-9e18, t + g.dist[v] + 1, -9e18, g.dist[v] + 1);
      }
      cout << res << '\n';
    } else
      assert(false);
  }
}
0