結果
問題 |
No.649 ここでちょっとQK!
|
ユーザー |
|
提出日時 | 2023-06-27 21:57:20 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 288 ms / 3,000 ms |
コード長 | 2,927 bytes |
コンパイル時間 | 689 ms |
コンパイル使用メモリ | 78,756 KB |
実行使用メモリ | 12,800 KB |
最終ジャッジ日時 | 2024-07-04 14:29:38 |
合計ジャッジ時間 | 6,265 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 32 |
ソースコード
/* k番目に小さい値を管理するデータ構造 DynamicPriorityQueue できること ・要素の追加 O(logN) dq.add(x) ・k番目に小さい要素の削除 O(1) dq.remove() ・任意の値を探して削除 0(logN) ・k番目に小さい値の出力 O(1) そもそもk個も存在しない場合 : -1を出力 k番目に大きい値であれば全部にマイナスをつければ良い コードチェックは https://yukicoder.me/problems/no/649 */ #include <iostream> #include <set> #include <vector> #include <algorithm> 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); } } void remove(ll value) { auto ite_small = find(small.begin(), small.end(), value); auto ite_large = find(large.begin(), large.end(), value); if(ite_small != small.end()) { small.erase(ite_small); if (!large.empty()) { ll next_kth = *large.begin(); large.erase(large.begin()); small.insert(next_kth); } } else if(ite_large != large.end()) { large.erase(ite_large); } else return; //どこにもなかったら何もしない } multiset<ll> k_small_set(){ return small; } }; 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; }