結果

問題 No.2809 Sort Query
ユーザー areik
提出日時 2024-07-12 23:14:30
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,117 ms / 2,000 ms
コード長 2,290 bytes
コンパイル時間 2,466 ms
コンパイル使用メモリ 135,276 KB
最終ジャッジ日時 2025-02-22 05:24:13
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 71
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <atcoder/lazysegtree.hpp>
#include <atcoder/segtree.hpp>
#include <cassert>
#include <cmath>
#include <cstring>
#include <deque>
#include <iomanip>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <tuple>
#include <utility>
#include <vector>
using namespace std;
using i8 = int8_t;
using i32 = int;
using i64 = long long;
// using i128 = __int128;
using u64 = unsigned long long;
using ll = long long;

void _main();
int main() {
  cin.tie(0);
  ios::sync_with_stdio(false);
  _main();
}

i64 pow(i64 j, i64 n) {
  i64 res = 1;
  while (n > 0) {
    if (n & 1)
      res *= j;
    j *= j;
    n >>= 1;
  }
  return res;
}

i64 op(i64 a, i64 b) { return a + b; }
i64 e() { return 0; }

i64 n, q;
void _main() {
  cin >> n >> q;
  vector<pair<i64, i64>> vs;
  vector<i64> a(n);
  for (i64 i = 0; i < n; i++) {
    cin >> a[i];
    vs.push_back({a[i], i});
    vs.push_back({0, i});
  }
  vector<tuple<i64, i64, i64>> query(q);
  for (i64 i = 0; i < q; i++) {
    i64 op, k, x;
    cin >> op;
    if (op == 1) {
      cin >> k >> x;
      vs.push_back({x, n + i});
    } else if (op == 3) {
      cin >> k;
    }
    k--;
    query[i] = {op, k, x};
  }
  sort(vs.begin(), vs.end());
  atcoder::segtree<i64, op, e> seg(vs.size());
  vector<i64> update(vs.size(), 0);
  set<i64> bk;
  for (i64 i = 0; i < n; i++) {
    seg.set(i, 1);
    i64 j = lower_bound(vs.begin(), vs.end(), make_pair(a[i], i)) - vs.begin();
    update[i] = j;
    bk.insert(i);
  }
  for (i64 i = 0; i < q; i++) {
    i64 op = get<0>(query[i]);
    i64 k = get<1>(query[i]);
    i64 x = get<2>(query[i]);
    if (op == 1) {
      i64 j =
          lower_bound(vs.begin(), vs.end(), make_pair(x, n + i)) - vs.begin();
      i64 p = seg.max_right(0, [&](i64 a) { return a <= k; });
      update[p] = j;
      bk.insert(p);
    } else if (op == 2) {
      while (!bk.empty()) {
        i64 j = *bk.begin();
        bk.erase(j);
        seg.set(j, seg.get(j) - 1);
        seg.set(update[j], seg.get(update[j]) + 1);
        update[j] = 0;
      }
    } else {
      i64 p = seg.max_right(0, [&](i64 a) { return a <= k; });
      if (update[p] == 0) {
        cout << vs[p].first << "\n";
      } else {
        cout << vs[update[p]].first << "\n";
      }
    }
  }
}
0