結果

問題 No.2338 Range AtCoder Query
ユーザー niuezniuez
提出日時 2023-06-02 23:02:22
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 4,760 bytes
コンパイル時間 1,643 ms
コンパイル使用メモリ 114,664 KB
実行使用メモリ 62,084 KB
最終ジャッジ日時 2024-06-09 01:00:03
合計ジャッジ時間 21,786 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 WA -
testcase_03 AC 2 ms
5,376 KB
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 AC 700 ms
62,084 KB
testcase_32 AC 647 ms
48,052 KB
testcase_33 WA -
testcase_34 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
#include <deque>
#include <cmath>

using namespace std;
using i64 = long long;
#define rep(i,s,e) for(i64 (i) = (s);(i) < (e);(i)++)
#define all(x) x.begin(),x.end()
#define STRINGIFY(n) #n
#define TOSTRING(n) STRINGIFY(n)
#define PREFIX "#" TOSTRING(__LINE__) "| "
#define debug(x) \
{ \
  std::cout << PREFIX << #x << " = " << x << std::endl; \
}
std::ostream& output_indent(std::ostream& os, int ind) {
  for(int i = 0; i < ind; i++) os << " ";
  return os;
}
template<class S, class T> std::ostream& operator<<(std::ostream& os, const std::pair<S, T>& p);
template<class T> std::ostream& operator<<(std::ostream& os, const std::vector<T>& v);
template<class S, class T> std::ostream& operator<<(std::ostream& os, const std::pair<S, T>& p) {
  return (os << "(" << p.first << ", " << p.second << ")");
}
template<class T> std::ostream& operator<<(std::ostream& os, const std::vector<T>& v) {
  os << "[";
  for(int i = 0;i < v.size();i++) os << v[i] << ", ";
  return (os << "]");
}
template<class T>
static inline std::vector<T> ndvec(size_t&& n, T val) { return std::vector<T>(n, std::forward<T>(val)); }
template<class... Tail>
static inline auto ndvec(size_t&& n, Tail&&... tail) {
  return std::vector<decltype(ndvec(std::forward<Tail>(tail)...))>(n, ndvec(std::forward<Tail>(tail)...));
}

#include <vector>
using i64 = long long;

template<class T, T (*ide)(), T (*ope)(T, T)>
struct segment_tree {
  i64 n;
  std::vector<T> node;

  segment_tree() {}
  segment_tree(const std::vector<T>& init) {
    n = 1;
    while(n < init.size()) n *= 2;
    node.resize(2 * n, ide());
    for(int i = 0;i < init.size();i++) node[i + n] = init[i];
    for(int i = n; i --> 1;) node[i] = ope(node[i * 2], node[i * 2 + 1]);
  }
 
  void update(i64 i, T x) {
    i += n;
    node[i] = x;
    while(i > 1) {
      i >>= 1;
      node[i] = ope(node[i * 2], node[i * 2 + 1]);
    }
  }

  T sum(i64 l, i64 r) const {
    T lx = ide();
    T rx = ide();
    l += n;
    r += n;
    while(l < r) {
      if(l & 1) { lx = ope(lx, node[l++]); }
      if(r & 1) { rx = ope(node[--r], rx); }
      l >>= 1;
      r >>= 1;
    }
    return ope(lx, rx);
  }

  const T& at(i64 i) const {
    return node[i + n];
  }

  template<class F>
  i64 subtree_down_first(i64 i, T lx, F isok) const {
    while(i < n) {
      T next = ope(lx, node[i * 2]);
      if(isok(next)) i = i * 2;
      else {
        lx = next;
        i = i * 2 + 1;
      }
    }
    return i - n + 1;
  }

  template<class F>
  i64 find_first(i64 l, F isok) const {
    if(isok(ide())) {
      return l;
    }
    if(l == 0) {
      if(isok(node[1])) return subtree_down_first(1, ide(), isok);
      return -1;
    }
    T lx = ide();
    i64 r = n << 1;
    l += n;
    while(l < r) {
      if(l & 1) {
        T next = ope(lx, node[l]);
        if(isok(next)) return subtree_down_first(l, lx, isok);
        lx = next;
        l++;
      }
      l >>= 1;
      r >>= 1;
    }
    return -1;
  }
};

int a_i() {
  return 0;
}

int a_o(int a, int b) {
  return a + b;
}

using Seg = segment_tree<int, a_i, a_o>;

int main() {
  int N, M, Q;
  cin >> N >> M >> Q;
  std::vector<i64> P, S;
  P.resize(N);
  S.resize(N);
  for(int i = 0; i < N; i++) {
    int p;
    string s;
    cin >> p >> s;
    P[i] = p - 1;
    S[i] = s == "AC" ? 1 : 0;
  }
  std::vector<std::vector<std::pair<int, int>>> query(N);
  for(int i = 0; i < Q; i++) {
    int a, b;
    cin >> a >> b;
    a--;
    b--;
    query[a].push_back({ b, i });
  }
  vector<int> before(M, -1);
  vector<vector<vector<int>>> was(M, vector<vector<int>>(1));
  vector<vector<int>> ac(M);
  vector<vector<int>> acs(N);
  vector<int> wi(M, 0);

  for(int i = 0; i < N; i++) {
    int p = P[i];
    if(S[i] == 0) {
      was[p].back().push_back(i);
    }
    else {
      ac[p].push_back(i);
      acs[i].push_back(p);
      was[p].push_back({});
    }
  }
  Seg seg_a(std::vector<int>(N, 0));
  Seg seg_w(std::vector<int>(N, 0));

  for(int p = 0; p < M; p++) {
    was[p].pop_back();
    if(wi[p] < was[p].size()) {
      seg_a.update(ac[p][wi[p]], 1);
      for(auto i: was[p][wi[p]]) {
        seg_w.update(i, 1);
      }
    }
  }

  std::vector<std::pair<int, int>> ans(Q);
  for(int i = 0; i < N; i++) {
    for(auto [r, q]: query[i]) {
      ans[q] = { seg_a.sum(i, r + 1), seg_w.sum(i, r + 1) };
    }

    for(auto p: acs[i]) {
      seg_a.update(ac[p][wi[p]], 0);
      for(auto i: was[p][wi[p]]) {
        seg_w.update(i, 0);
      }
      wi[p]++;
      if(wi[p] < was[p].size()) {
        seg_a.update(ac[p][wi[p]], 1);
        for(auto i: was[p][wi[p]]) {
          seg_w.update(i, 1);
        }
      }
    }
  }
  for(auto [a, b]: ans) {
    cout << a << " " << b << endl;
  }
}
0