結果
問題 | No.649 ここでちょっとQK! |
ユーザー |
|
提出日時 | 2021-01-31 18:36:33 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 98 ms / 3,000 ms |
コード長 | 2,856 bytes |
コンパイル時間 | 1,573 ms |
コンパイル使用メモリ | 140,616 KB |
最終ジャッジ日時 | 2025-01-18 10:20:01 |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 32 |
ソースコード
#include <iostream>#include <vector>#include <algorithm>#include <cmath>#include <queue>#include <string>#include <map>#include <set>#include <stack>#include <tuple>#include <deque>#include <array>#include <numeric>#include <bitset>#include <iomanip>#include <cassert>#include <chrono>#include <random>#include <limits>#include <iterator>#include <functional>#include <sstream>#include <fstream>#include <complex>#include <cstring>#include <unordered_map>#include <unordered_set>using namespace std;using ll = long long;constexpr int INF = 1001001001;constexpr int mod = 1000000007;// constexpr int mod = 998244353;template<class T>inline bool chmax(T& x, T y){if(x < y){x = y;return true;}return false;}template<class T>inline bool chmin(T& x, T y){if(x > y){x = y;return true;}return false;}template<typename T>struct BinaryIndexedTree{vector<T> data;int sz;BinaryIndexedTree(int n){sz = n + 2;data.assign(sz, 0);}void add(int i, T x){++i;while(i < sz){data[i] += x;i += i & -i;}}// [0, i] の区間和T sum(int i){T res = 0;++i;while(i > 0){res += data[i];i -= i & -i;}return res;}// [l, r) の区間和T sum(int l, int r){if(l > r) return 0;return sum(r - 1) - sum(l - 1);}// val 以上を満たす 0-indexed の位置を返すint lower_bound(T val){// if(val <= 0) return -1;int res = 0;int n = 1;while(n < sz) n <<= 1;for(int k = n >> 1; k > 0; k >>= 1){if(res + k < sz && data[res + k] < val){val -= data[res + k];res += k;}}return res; // 1-indexed であれば、res + 1 を返す}};int main(){ios::sync_with_stdio(false);cin.tie(nullptr);int Q, K;cin >> Q >> K;vector<pair<int, ll>> querys(Q);vector<ll> ys;for(auto& [x, y] : querys){cin >> x;if(x == 1){cin >> y;ys.emplace_back(y);}}sort(ys.begin(), ys.end());ys.erase(unique(ys.begin(), ys.end()), ys.end());int Y = ys.size();BinaryIndexedTree<int> bit(Y);for(auto& [x, y] : querys){if(x == 1){int i = lower_bound(ys.begin(), ys.end(), y) - ys.begin();bit.add(i, 1);}else{if(bit.sum(0, Y) < K){cout << -1 << '\n';}else{int i = bit.lower_bound(K);cout << ys[i] << '\n';bit.add(i, -1);}}}return 0;}