結果
問題 |
No.649 ここでちょっとQK!
|
ユーザー |
|
提出日時 | 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 |
ソースコード
#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; } } } }