結果

問題 No.1216 灯籠流し/Lanterns
ユーザー 👑 hos.lyrichos.lyric
提出日時 2020-08-30 16:00:58
言語 C++14
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 1,727 ms / 4,500 ms
コード長 6,581 bytes
コンパイル時間 1,780 ms
コンパイル使用メモリ 135,028 KB
実行使用メモリ 146,476 KB
最終ジャッジ日時 2023-08-09 14:45:01
合計ジャッジ時間 38,714 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge11
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,376 KB
testcase_01 AC 1 ms
4,380 KB
testcase_02 AC 1 ms
4,384 KB
testcase_03 AC 1 ms
4,376 KB
testcase_04 AC 42 ms
19,920 KB
testcase_05 AC 1,039 ms
106,088 KB
testcase_06 AC 57 ms
12,832 KB
testcase_07 AC 1,067 ms
93,400 KB
testcase_08 AC 938 ms
71,936 KB
testcase_09 AC 385 ms
44,088 KB
testcase_10 AC 444 ms
43,676 KB
testcase_11 AC 152 ms
29,808 KB
testcase_12 AC 34 ms
16,264 KB
testcase_13 AC 207 ms
39,168 KB
testcase_14 AC 85 ms
25,068 KB
testcase_15 AC 166 ms
34,896 KB
testcase_16 AC 138 ms
38,836 KB
testcase_17 AC 738 ms
74,920 KB
testcase_18 AC 194 ms
37,280 KB
testcase_19 AC 1,087 ms
131,360 KB
testcase_20 AC 439 ms
69,404 KB
testcase_21 AC 213 ms
43,656 KB
testcase_22 AC 901 ms
96,888 KB
testcase_23 AC 57 ms
25,924 KB
testcase_24 AC 63 ms
12,484 KB
testcase_25 AC 31 ms
9,072 KB
testcase_26 AC 500 ms
44,284 KB
testcase_27 AC 125 ms
26,844 KB
testcase_28 AC 77 ms
24,680 KB
testcase_29 AC 213 ms
38,964 KB
testcase_30 AC 353 ms
46,752 KB
testcase_31 AC 872 ms
66,140 KB
testcase_32 AC 334 ms
48,660 KB
testcase_33 AC 67 ms
20,224 KB
testcase_34 AC 1,095 ms
110,308 KB
testcase_35 AC 1,727 ms
127,228 KB
testcase_36 AC 1,220 ms
81,916 KB
testcase_37 AC 1,026 ms
115,624 KB
testcase_38 AC 1,256 ms
113,876 KB
testcase_39 AC 1,299 ms
108,844 KB
testcase_40 AC 1,522 ms
128,720 KB
testcase_41 AC 1,503 ms
129,064 KB
testcase_42 AC 1,548 ms
129,164 KB
testcase_43 AC 1,496 ms
129,244 KB
testcase_44 AC 1,518 ms
129,412 KB
testcase_45 AC 1,253 ms
146,456 KB
testcase_46 AC 1,272 ms
146,476 KB
testcase_47 AC 1,274 ms
146,256 KB
testcase_48 AC 1,252 ms
146,452 KB
testcase_49 AC 1,242 ms
145,796 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cassert>
#include <cmath>
#include <cstdint>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <bitset>
#include <complex>
#include <deque>
#include <functional>
#include <iostream>
#include <map>
#include <numeric>
#include <queue>
#include <set>
#include <sstream>
#include <string>
#include <unordered_map>
#include <unordered_set>
#include <utility>
#include <vector>

using namespace std;

using Int = long long;

template <class T1, class T2> ostream &operator<<(ostream &os, const pair<T1, T2> &a) { return os << "(" << a.first << ", " << a.second << ")"; };
template <class T> void pv(T a, T b) { for (T i = a; i != b; ++i) cerr << *i << " "; cerr << endl; }
template <class T> bool chmin(T &t, const T &f) { if (t > f) { t = f; return true; } return false; }
template <class T> bool chmax(T &t, const T &f) { if (t < f) { t = f; return true; } return false; }

int N, Q;
vector<int> A, B;
vector<Int> C;
vector<int> TYP, V;
vector<Int> T, L;

vector<vector<int>> G;
vector<int> par, sz, dep;
vector<Int> dist;
int O;
vector<int> ord, head;

void dfs0(int u, int p, int ip) {
  par[u] = p;
  sz[u] = 1;
  dep[u] = (p == -1) ? 0 : (dep[p] + 1);
  dist[u] = (p == -1) ? 0 : (dist[p] + C[ip]);
  for (const int i : G[u]) {
    const int v = A[i] ^ B[i] ^ u;
    if (v != p) {
      dfs0(v, u, i);
      sz[u] += sz[v];
    }
  }
}

void dfs1(int u, int p, int h) {
  ord[u] = O++;
  head[u] = h;
  int vm = -1;
  for (const int i : G[u]) {
    const int v = A[i] ^ B[i] ^ u;
    if (v != p) {
      if (vm == -1 || sz[vm] < sz[v]) {
        vm = v;
      }
    }
  }
  if (vm != -1) {
    dfs1(vm, u, h);
    for (const int i : G[u]) {
      const int v = A[i] ^ B[i] ^ u;
      if (v != p && v != vm) {
        dfs1(v, u, v);
      }
    }
  }
}

int BIT_N;
struct BIT {
  map<int, int> as;
  void add(int pos, int val) {
    for (int x = pos; x < BIT_N; x |= x + 1) {
      as[x] += val;
    }
  }
  int sum(int pos) {
    int ret = 0;
    for (int x = pos - 1; x >= 0; x = (x & (x + 1)) - 1) {
      auto it = as.find(x);
      if (it != as.end()) {
        ret += it->second;
      }
    }
    return ret;
  }
};

constexpr int E = 16;

int main() {
  for (; ~scanf("%d%d", &N, &Q); ) {
    A.resize(N - 1);
    B.resize(N - 1);
    C.resize(N - 1);
    for (int i = 0; i < N - 1; ++i) {
      scanf("%d%d%lld", &A[i], &B[i], &C[i]);
      --A[i];
      --B[i];
    }
    TYP.resize(Q);
    V.resize(Q);
    T.resize(Q);
    L.resize(Q);
    for (int q = 0; q < Q; ++q) {
      scanf("%d%d%lld%lld", &TYP[q], &V[q], &T[q], &L[q]);
      --V[q];
    }
    
    G.assign(N, {});
    for (int i = 0; i < N - 1; ++i) {
      G[A[i]].push_back(i);
      G[B[i]].push_back(i);
    }
    
    par.resize(N);
    sz.resize(N);
    dep.resize(N);
    dist.resize(N);
    constexpr int rt = 0;
    dfs0(rt, -1, -1);
    O = 0;
    ord.resize(N);
    head.resize(N);
    dfs1(rt, -1, rt);
// cerr<<"dist = ";pv(dist.begin(),dist.end());
// cerr<<"ord = ";pv(ord.begin(),ord.end());
// cerr<<"head = ";pv(head.begin(),head.end());
    
    vector<int> lens(N, 0);
    for (int u = 0; u < N; ++u) {
      ++lens[head[u]];
    }
    BIT_N = Q;
    vector<vector<BIT>> bits(N);
    for (int h = 0; h < N; ++h) {
      bits[h].assign(lens[h], {});
    }
    
    auto bAdd = [&](int h, int pos, int key, int val) {
// cerr<<"bAdd "<<h<<" "<<pos<<" "<<key<<" "<<val<<endl;
      for (int x = pos; x < lens[h]; x |= x + 1) {
        bits[h][x].add(key, val);
      }
    };
    auto bSum = [&](int h, int pos, int key) {
      int ret = 0;
      for (int x = pos - 1; x >= 0; x = (x & (x + 1)) - 1) {
        ret += bits[h][x].sum(key);
      }
      return ret;
    };
    
    // the path from u to root
    auto hldAdd = [&](int u, int key, int val) {
// cerr<<"hldAdd "<<u<<" "<<key<<" "<<val<<endl;
      for (; u != -1; ) {
        const int h = head[u];
        // add to [0, ord[u] - ord[h]]
// cerr<<"  "<<h<<" "<<(lens[h] - (ord[u] - ord[h]) - 1)<<endl;
        bAdd(h, lens[h] - (ord[u] - ord[h]) - 1, key, val);
        u = par[h];
      }
    };
    auto hldSum = [&](int u, int key) {
// cerr<<"hldSum "<<u<<" "<<key<<endl;
      const int h = head[u];
      // sum at (ord[u] - ord[h])
// cerr<<"  "<<h<<" "<<(lens[h] - (ord[u] - ord[h]))<<endl;
      const int ret = bSum(h, lens[h] - (ord[u] - ord[h]), key);
      return ret;
    };
    
    vector<vector<int>> pp(E, vector<int>(N));
    vector<vector<Int>> dd(E, vector<Int>(N));
    pp[0] = par;
    for (int u = 0; u < N; ++u) {
      dd[0][u] = (par[u] == -1) ? 0 : (dist[u] - dist[par[u]]);
    }
    for (int e = 0; e < E - 1; ++e) {
      for (int u = 0; u < N; ++u) {
        if (pp[e][u] == -1) {
          pp[e + 1][u] = -1;
          dd[e + 1][u] = dd[e][u];
        } else {
          pp[e + 1][u] = pp[e][pp[e][u]];
          dd[e + 1][u] = dd[e][u] + dd[e][pp[e][u]];
        }
      }
    }
    
/*
vector<int>brt(Q,-58);
if(N<=2000&&Q<=2000){
 vector<vector<Int>>wss(N);
 for(int q=0;q<Q;++q){
  if(TYP[q]==0){
   for(int u=V[q];u!=-1&&dist[V[q]]-dist[u]<=L[q];u=par[u]){
// cerr<<"brt add "<<u<<" "<<T[q]+dist[V[q]]<<endl;
    wss[u].push_back(T[q]+dist[V[q]]);
   }
  }else{
   sort(wss[V[q]].begin(),wss[V[q]].end());
   brt[q]=upper_bound(wss[V[q]].begin(),wss[V[q]].end(),T[q]+dist[V[q]])-wss[V[q]].begin();
   // printf("%d\n",brt[q]);
  }
 }
 // continue;
}
// exit(1);
//*/
    
    vector<int> ans(Q, -58);
    vector<pair<pair<Int, int>, int>> qrys(Q);
    for (int q = 0; q < Q; ++q) {
      qrys[q] = make_pair(make_pair(T[q] + dist[V[q]], TYP[q]), q);
    }
    sort(qrys.begin(), qrys.end());
    for (const auto &qry : qrys) {
      const int q = qry.second;
      switch (TYP[q]) {
        case 0: {
          Int l = L[q];
          int u = V[q];
          for (int e = E; e--; ) {
            if (pp[e][u] != -1 && l - dd[e][u] >= 0) {
              l -= dd[e][u];
              u = pp[e][u];
            }
          }
// cerr<<"q = "<<q<<": u = "<<u<<endl;
assert(dist[V[q]]-dist[u]<=L[q]);
if(par[u]!=-1)assert(dist[V[q]]-dist[par[u]]>L[q]);
          hldAdd(V[q], q, +1);
          hldAdd(par[u], q, -1);
        } break;
        case 1: {
// cerr<<"q = "<<q<<endl;
          ans[q] = hldSum(V[q], q);
        } break;
        default: assert(false);
      }
    }
    
    for (int q = 0; q < Q; ++q) {
      if (TYP[q] == 1) {
        printf("%d\n", ans[q]);
      }
    }
/*
if(N<=2000&&Q<=2000){
 for(int q=0;q<Q;++q)if(brt[q]!=ans[q]){
  cerr<<"q = "<<q<<": "<<brt[q]<<" "<<ans[q]<<endl;
  assert(false);
 }
}
//*/
  }
  return 0;
}
0