TakahiroNakamori

TakahiroNakamori

ナカモリタカヒロ

 
 

Sequence Query

問題

D – Sequence Query
プログラミング初級者から上級者まで楽しめる、競技プログラミングコンテストサイト「AtCoder」。オンラインで毎週開催プログラミングコンテストを開催しています。競技…

記録

解法

コード例

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using ll = long long;
using P = pair<ll, ll>;
#define rep(i, n) for(int i = 0; i < (n); ++i)
// using mint = modint998244353;
// using mint = modint1000000007;
// const int mod = 1000000007;
// const ll INF = 1LL << 62;
// const int INF = 1001001001;

int main() {
  int Q;
  cin >> Q;

  multiset<ll> A;

  while(0 < Q) {
    int t;
    cin >> t;
    ll x;
    cin >> x;
    if(t == 1) {
      A.insert(x);
    } else if(t == 2) {
      int k;
      cin >> k;
      auto res = A.upper_bound(x);
      ll ans = 0;
      rep(i, k) {
        if(res == A.begin()) {
          ans = -1;
          break;
        }
        res = prev(res);
      }
      if(ans == 0) {
        ans = *res;
      }
      cout << ans << endl;
    } else {
      int k;
      cin >> k;
      k--;
      auto res = A.lower_bound(x);
      ll ans = 0;
      rep(i, k) {
        if(res == A.end()) {
          ans = -1;
          break;
        }
        res = next(res);
      }
      if(res == A.end()) {
        ans = -1;
      }
      if(ans == 0) {
        ans = *res;
      }
      cout << ans << endl;
    }
    Q--;
  }

  return 0;
}