結果

問題 No.649 ここでちょっとQK！
ユーザー hashiryohashiryo
提出日時 2020-04-28 23:49:15
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 270 ms / 3,000 ms
コード長 1,863 bytes
コンパイル時間 1,805 ms
コンパイル使用メモリ 176,036 KB
実行使用メモリ 8,132 KB
最終ジャッジ日時 2024-05-04 13:39:08
合計ジャッジ時間 7,166 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 270 ms
5,476 KB
testcase_04 AC 72 ms
8,128 KB
testcase_05 AC 72 ms
8,132 KB
testcase_06 AC 56 ms
6,600 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 62 ms
5,500 KB
testcase_13 AC 62 ms
5,496 KB
testcase_14 AC 62 ms
5,628 KB
testcase_15 AC 62 ms
5,500 KB
testcase_16 AC 62 ms
5,756 KB
testcase_17 AC 68 ms
5,756 KB
testcase_18 AC 73 ms
6,012 KB
testcase_19 AC 81 ms
6,272 KB
testcase_20 AC 88 ms
6,272 KB
testcase_21 AC 99 ms
6,712 KB
testcase_22 AC 103 ms
6,836 KB
testcase_23 AC 109 ms
7,088 KB
testcase_24 AC 115 ms
7,220 KB
testcase_25 AC 122 ms
7,608 KB
testcase_26 AC 128 ms
7,984 KB
testcase_27 AC 2 ms
5,376 KB
testcase_28 AC 2 ms
5,376 KB
testcase_29 AC 2 ms
5,376 KB
testcase_30 AC 53 ms
5,376 KB
testcase_31 AC 53 ms
5,376 KB
testcase_32 AC 2 ms
5,376 KB
testcase_33 AC 2 ms
5,376 KB
testcase_34 AC 2 ms
5,376 KB
testcase_35 AC 2 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

#define debug(x) cerr << #x << ": " << x << endl
#define debugArray(x, n)                           \
  for (long long hoge = 0; (hoge) < (n); ++(hoge)) \
  cerr << #x << "[" << hoge << "]: " << x[hoge] << endl

struct BinaryIndexedTree {
  vector<long long> dat;
  BinaryIndexedTree(int n) : dat(n + 1, 0) {}
  BinaryIndexedTree(int n, long long a)
      : BinaryIndexedTree(vector<long long>(n, a)) {}
  BinaryIndexedTree(vector<long long> y) : dat(y.size() + 1) {
    for (int k = 0; k < y.size(); ++k) dat[k + 1] = y[k];
    for (int k = 1; k + (k & -k) < dat.size(); ++k) dat[k + (k & -k)] += dat[k];
  }
  void add(int k, long long a) {
    for (++k; k < dat.size(); k += k & -k) dat[k] += a;
  }
  // sum [0,k)
  long long operator[](int k) {
    long long s = 0;
    for (; k > 0; k &= k - 1) s += dat[k];
    return s;
  }
  // min{ k : sum(k) >= a }
  int lower_bound(long long a) const {
    int k = 0;
    for (int p = 1 << (__lg(dat.size() - 1) + 1); p > 0; p >>= 1)
      if (k + p < dat.size() && dat[k + p] < a) a -= dat[k += p];
    return k + 1 == dat.size() ? -1 : k;
  }
};

signed main() {
  cin.tie(0);
  ios::sync_with_stdio(0);
  int Q, K;
  cin >> Q >> K;
  vector<long long> query;
  vector<long long> x;
  while (Q--) {
    long long v;
    cin >> v;
    if (v == 1) {
      cin >> v;
      x.push_back(v);
    } else {
      v = -1;
    }
    query.push_back(v);
  }
  sort(x.begin(), x.end());
  x.erase(unique(x.begin(), x.end()), x.end());
  BinaryIndexedTree bit(x.size());
  for (auto q : query) {
    if (q < 0) {
      int i = bit.lower_bound(K);
      if (i >= 0) {
        cout << x[i] << endl;
        bit.add(i, -1);
      } else {
        cout << -1 << endl;
      }
    } else {
      int i = lower_bound(x.begin(), x.end(), q) - x.begin();
      bit.add(i, 1);
    }
  }
}
0