結果
問題 |
No.649 ここでちょっとQK!
|
ユーザー |
|
提出日時 | 2023-06-27 18:33:31 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 287 ms / 3,000 ms |
コード長 | 1,783 bytes |
コンパイル時間 | 734 ms |
コンパイル使用メモリ | 75,776 KB |
実行使用メモリ | 12,800 KB |
最終ジャッジ日時 | 2024-07-04 11:07:56 |
合計ジャッジ時間 | 6,234 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 32 |
ソースコード
#include <iostream> #include <set> #include <vector> using namespace std; using ll = long long; using vll = vector<ll>; struct DynamicMultiset { multiset<ll> small; // smallは昇順で要素を保持するmultiset multiset<ll> large; // largeは昇順で要素を保持するmultiset int K; // k番目に小さい値の管理数 void add(ll value) { if (small.size() < K) { small.insert(value); } else { ll current_kth = *(--small.end()); if (value < current_kth) { small.erase(--small.end()); small.insert(value); large.insert(current_kth); } else { large.insert(value); } } } ll getKthSmallest() { if (small.size() < K) { return -1; } return *(--small.end()); } void remove() { if (small.size() < K) { return; } ll current_kth = *(--small.end()); small.erase(--small.end()); if (!large.empty()) { ll next_kth = *large.begin(); large.erase(large.begin()); small.insert(next_kth); } } }; using ll = long long; using vll = vector<long long>; int main() { DynamicMultiset dm; int Q, K; cin >> Q >> K; dm.K = K; vll ans; for (int i = 0; i < Q; ++i) { int T; cin >> T; if (T == 1) { ll V; cin >> V; dm.add(V); } else { ll kthSmallest = dm.getKthSmallest(); ans.push_back(kthSmallest); dm.remove(); } } for (int i = 0; i < ans.size(); ++i) { cout << ans[i] << endl; } return 0; }