結果

問題 No.649 ここでちょっとQK！
ユーザー ei1333333
提出日時 2018-02-09 23:16:10
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 113 ms / 3,000 ms
コード長 1,606 bytes
コンパイル時間 2,263 ms
コンパイル使用メモリ 201,808 KB
最終ジャッジ日時 2025-01-05 08:16:05
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 32
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:61:8: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   61 |   scanf("%d %d", &Q, &K);
      |   ~~~~~^~~~~~~~~~~~~~~~~
main.cpp:64:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   64 |     scanf("%d", &T[i]);
      |     ~~~~~^~~~~~~~~~~~~
main.cpp:66:12: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   66 |       scanf("%lld", &V[i]);
      |       ~~~~~^~~~~~~~~~~~~~~

ソースコード

diff # 

#include<bits/stdc++.h>

using namespace std;


using int64 = long long;

template< class T >
struct BinaryIndexedTree
{
  vector< T > data;

  BinaryIndexedTree(int sz)
  {
    data.assign(++sz, 0);
  }

  T sum(int k)
  {
    T ret = 0;
    for(++k; k > 0; k -= k & -k) ret += data[k];
    return (ret);
  }

  void add(int k, T x)
  {
    for(++k; k < data.size(); k += k & -k) data[k] += x;
  }
};

template< class T >
struct AdditionalBIT : BinaryIndexedTree< T >
{
  int curr;

  AdditionalBIT(int sz) : BinaryIndexedTree< T >(sz)
  {
    curr = 1;
    while(curr <= sz) curr <<= 1;
  }

  int lower_bound(T w)
  {
    if(w <= 0) return (0);
    int i = 0;
    for(int k = curr; k > 0; k >>= 1) {
      if(i + k < BinaryIndexedTree< T >::data.size() && BinaryIndexedTree< T >::data[i + k] < w) {
        w -= BinaryIndexedTree< T >::data[i + k];
        i += k;
      }
    }
    return (i);
  }
};

int main()
{
  int Q, K, T[200000];
  int64 V[200000];

  scanf("%d %d", &Q, &K);
  vector< int64 > vs;
  for(int i = 0; i < Q; i++) {
    scanf("%d", &T[i]);
    if(T[i] == 1) {
      scanf("%lld", &V[i]);
      vs.push_back(V[i]);
    }
  }
  sort(begin(vs), end(vs));
  vs.erase(unique(begin(vs), end(vs)), end(vs));
  AdditionalBIT< int > bit(vs.size());
  int sz = 0;
  for(int i = 0; i < Q; i++) {
    if(T[i] == 1) {
      bit.add(lower_bound(begin(vs), end(vs), V[i]) - begin(vs), 1);
      ++sz;
    } else {
      if(sz < K) {
        puts("-1");
      } else {
        auto p = bit.lower_bound(K);
        printf("%lld\n", vs[p]);
        bit.add(p, -1);
        --sz;
      }
    }
  }
}
0