/* 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 #include #include #include using namespace std; using ll = long long; using vll = vector; struct DynamicMultiset { multiset small; // smallは昇順で要素を保持するmultiset multiset 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 = small.find(value); auto ite_large = large.find(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 k_small_set(){ return small; } }; using ll = long long; using vll = vector; 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(kthSmallest); } } for (int i = 0; i < ans.size(); ++i) { cout << ans[i] << endl; } return 0; }