結果

問題 No.649 ここでちょっとQK！
コンテスト
ユーザー Davaajav JARGALSAIKHAN
提出日時 2021-01-14 23:52:11
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 283 ms / 3,000 ms
コード長 2,242 bytes
コンパイル時間 1,212 ms
コンパイル使用メモリ 105,140 KB
実行使用メモリ 8,716 KB
最終ジャッジ日時 2024-11-24 19:05:53
合計ジャッジ時間 6,479 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <vector>
#include <list>
#include <map>
#include <set>
#include <queue>
#include <deque>
#include <stack>
#include <bitset>
#include <algorithm>
#include <functional>
#include <numeric>
#include <utility>
#include <sstream>
#include <iostream>
#include <iomanip>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <ctime>
#define sz(a) int((a).size())
#define rep(i,n) for(int i=0;i<(n);++i)
#define ALL(c) c.begin(), c.end()
#define tr(container, it) for(auto it = container.begin(); it != container.end(); it++)
#define ZEROS(a) memset(a,0,sizeof(a))
#define MINUS(a) memset(a,-1,sizeof(a))

using namespace std;

typedef std::pair< int,int > P;
typedef long long ll;

struct BIT {
    int N;
    vector<int> data;
    BIT(int N): N(N), data(N+1, 0) {}
    void add (int x, int val) {
        while (x <= N) {
            data[x] += val;
            x += x & -x;
        }
    }

    ll sum (int x) {
        ll sum = 0;
        while (x > 0) {
            sum += data[x];
            x -= x & -x;
        }
        return sum;
    }

    ll sum_all() {
        return sum(N);
    } 

    int query (int k) {
        if (sum_all() < k) return -1;
        int lo = 0;
        int M = 1;
        while (M <= N) M*=2;
        for (int i = M/2; i>0; i/=2) {
            if (lo + i <= N && data[lo + i] < k) {
                k = k - data[lo + i];
                lo = lo + i;
            }
        }
        return lo + 1;
    } 
};

BIT bittree(200005);
vector<ll> values(200005, 0);
vector<int> queries(200005, 0);
int Q, K;

int main () {
    cin >> Q >> K;
    vector<ll> V;
    rep(i,Q) {
        cin >> queries[i];
        if (queries[i] == 1) cin >> values[i], V.push_back(values[i]);  
    }
    sort(ALL(V));
    V.erase( unique(ALL(V)), V.end() );
    rep(i,Q) {
        if(queries[i] == 1) {
            auto it = lower_bound( ALL(V), values[i] );
            int ind = it - V.begin();
            ind++;
            bittree.add(ind, 1);
        } else {
            int kth_elem = bittree.query(K);
            if (kth_elem == -1) {
                cout << -1 << endl;
            } else {
                bittree.add(kth_elem, -1);
                cout << V[--kth_elem] << endl;
            }
        }
    }
}
0