結果
問題 |
No.649 ここでちょっとQK!
|
ユーザー |
|
提出日時 | 2019-12-19 02:57:38 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 277 ms / 3,000 ms |
コード長 | 1,113 bytes |
コンパイル時間 | 816 ms |
コンパイル使用メモリ | 83,396 KB |
実行使用メモリ | 7,896 KB |
最終ジャッジ日時 | 2024-07-06 23:24:51 |
合計ジャッジ時間 | 6,026 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 32 |
コンパイルメッセージ
main.cpp:37:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type] 37 | main() | ^~~~
ソースコード
#include<iostream> #include<map> #include<algorithm> #include<vector> using namespace std; //1-indexed #include<vector> template<typename T> struct BIT{ int n; vector<T>bit; BIT(int n_=0):n(n_),bit(n_+1){} T sum(int i) { T ans=0; for(;i>0;i-=i&-i)ans+=bit[i]; return ans; } void add(int i,T a) { if(i==0)return; for(;i<=n;i+=i&-i)bit[i]+=a; } int lower_bound(T k)//k<=sum(ret) { if(k<=0)return 0; int ret=0,i=1; while((i<<1)<=n)i<<=1; for(;i;i>>=1) if(ret+i<=n&&bit[ret+i]<k)k-=bit[ret+=i]; return ret+1; } }; int Q,K; int q[2<<17]; long v[2<<17]; main() { cin>>Q>>K; vector<long>V; for(int i=0;i<Q;i++) { cin>>q[i]; if(q[i]==1) { cin>>v[i]; V.push_back(v[i]); } } sort(V.begin(),V.end()); V.erase(unique(V.begin(),V.end()),V.end()); BIT<int>P(V.size()); int cnt=0; for(int i=0;i<Q;i++) { if(q[i]==1) { int id=lower_bound(V.begin(),V.end(),v[i])-V.begin(); P.add(id+1,1); cnt++; } else { if(cnt<K)cout<<-1<<endl; else { int id=P.lower_bound(K); P.add(id,-1); cnt--; id--; cout<<V[id]<<endl; } } } }