結果

問題 No.2809 Sort Query
ユーザー silv723
提出日時 2024-04-21 20:22:01
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,927 bytes
コンパイル時間 18,336 ms
コンパイル使用メモリ 314,812 KB
最終ジャッジ日時 2025-02-21 08:25:05
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 53 WA * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC optimize("Ofast")
#pragma GCC target ("avx2")
#include<bits/stdc++.h>
using namespace std;
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace __gnu_pbds;
struct segtree{
  int n;
  vector<int> dat;
  segtree(int _n):n(_n){
    dat.resize(2*n);
  }
  void update(int i, int c){
    i += n;
    dat[i] = c;
    i /= 2;
    while(i){
      dat[i] = dat[2*i] + dat[2*i+1];
      i /= 2;
    }
  }
  int sum(int l, int r){
    l += n;
    r += n;
    int ret = 0;
    while(l < r){
      if(l & 1) ret += dat[l++];
      if(r & 1) ret += dat[--r];
      l /= 2, r /= 2;
    }
    return ret;
  }
};
int main(){
  int n, q;
  scanf("%d%d", &n, &q);
  assert(1 <= n && n <= 500000 && 1 <= q && q <= 500000);
  tree<long long, null_type, less_equal<long long>, rb_tree_tag, tree_order_statistics_node_update> m;
  segtree ch(n);
  gp_hash_table<int, long long> m2;
  for(int i = 0; i < n; i++){
    long long x;
    scanf("%ld", &x);
    assert(1 <= x && x <= 1000000000000000000LL);
    m.insert(x);
  }
  while(q--){
    int type;
    scanf("%d", &type);
    assert(1 <= type && type <= 3);
    if(type == 1){
      int k;
      long long x;
      scanf("%d%ld", &k, &x);
      assert(1 <= k && k <= n && 1 <= x && x <= 1000000000000000000LL);
      k--;
      if(m2.find(k) != m2.end()){
        m2[k] = x;
      }else{
        m2[k] = x;
        ch.update(k, 1);
        int c = ch.sum(0, k);
        auto p = m.find_by_order(k - c);
        m.erase(p);
      }
    }else if(type == 2){
      for(auto [p, q] : m2){
        ch.update(p, 0);
        m.insert(q);
      }
      m2.clear();
    }else{
      int k;
      scanf("%d", &k);
      assert(1 <= k && k <= n);
      k--;
      if(m2.find(k) != m2.end()){
        printf("%ld\n", m2[k]);
      }else{
        int c = ch.sum(0, k);
        auto p = m.find_by_order(k - c);
        printf("%ld\n", *p);
      }
    }
  }
}
0