結果

問題 No.649 ここでちょっとQK！
ユーザー Davaajav JARGALSAIKHANDavaajav JARGALSAIKHAN
提出日時 2021-01-14 23:52:11
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 304 ms / 3,000 ms
コード長 2,242 bytes
コンパイル時間 1,228 ms
コンパイル使用メモリ 104,960 KB
実行使用メモリ 8,840 KB
最終ジャッジ日時 2024-05-03 16:36:42
合計ジャッジ時間 6,656 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 4 ms
6,400 KB
testcase_01 AC 5 ms
6,528 KB
testcase_02 AC 4 ms
6,528 KB
testcase_03 AC 304 ms
6,400 KB
testcase_04 AC 188 ms
8,712 KB
testcase_05 AC 187 ms
8,712 KB
testcase_06 AC 186 ms
8,588 KB
testcase_07 AC 4 ms
6,400 KB
testcase_08 AC 4 ms
6,400 KB
testcase_09 AC 4 ms
6,528 KB
testcase_10 AC 4 ms
6,656 KB
testcase_11 AC 4 ms
6,528 KB
testcase_12 AC 117 ms
7,560 KB
testcase_13 AC 118 ms
7,560 KB
testcase_14 AC 114 ms
7,560 KB
testcase_15 AC 116 ms
7,508 KB
testcase_16 AC 115 ms
7,560 KB
testcase_17 AC 126 ms
7,560 KB
testcase_18 AC 138 ms
7,556 KB
testcase_19 AC 151 ms
7,692 KB
testcase_20 AC 160 ms
7,556 KB
testcase_21 AC 174 ms
8,584 KB
testcase_22 AC 185 ms
8,840 KB
testcase_23 AC 196 ms
8,580 KB
testcase_24 AC 208 ms
8,588 KB
testcase_25 AC 225 ms
8,712 KB
testcase_26 AC 230 ms
8,580 KB
testcase_27 AC 5 ms
6,528 KB
testcase_28 AC 4 ms
6,528 KB
testcase_29 AC 5 ms
6,400 KB
testcase_30 AC 81 ms
7,560 KB
testcase_31 AC 79 ms
7,560 KB
testcase_32 AC 4 ms
6,400 KB
testcase_33 AC 4 ms
6,400 KB
testcase_34 AC 4 ms
6,400 KB
testcase_35 AC 5 ms
6,400 KB
権限があれば一括ダウンロードができます

ソースコード

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