結果
問題 |
No.649 ここでちょっとQK!
|
ユーザー |
|
提出日時 | 2019-09-25 00:10:53 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 506 ms / 3,000 ms |
コード長 | 2,282 bytes |
コンパイル時間 | 1,849 ms |
コンパイル使用メモリ | 188,272 KB |
実行使用メモリ | 44,928 KB |
最終ジャッジ日時 | 2024-09-19 14:53:23 |
合計ジャッジ時間 | 9,883 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 32 |
ソースコード
#include"bits/stdc++.h" using namespace std; #define REP(k,m,n) for(int (k)=(m);(k)<(n);(k)++) #define rep(i,n) REP((i),0,(n)) using ll = long long; template<typename T> class SegmentTree { private: using F = function<T(T, T)>; // モノイド型 int n; // 横幅 F f; // モノイド T e; // モノイド単位元 vector<T> data; public: // init忘れに注意 SegmentTree() {} SegmentTree(F f, T e) :f(f), e(e) {} void init(int n_) { n = 1; while (n < n_)n <<= 1; data.assign(n << 1, e); } void build(const vector<T>& v) { int n_ = v.size(); init(n_); rep(i, n_)data[n + i] = v[i]; for (int i = n - 1; i >= 0; i--) { data[i] = f(data[(i << 1) | 0], data[(i << 1) | 1]); } } void set_val(int idx, T val) { idx += n; data[idx] = val; while (idx >>= 1) { data[idx] = f(data[(idx << 1) | 0], data[(idx << 1) | 1]); } } T query(int a, int b) { // [a,b) T vl = e, vr = e; for (int l = a + n, r = b + n; l < r; l >>= 1, r >>= 1) { if (l & 1)vl = f(vl, data[l++]); // unknown if (r & 1)vr = f(data[--r], vr); // unknown } return f(vl, vr); } }; int main() { // input ll Q, K; cin >> Q >> K; vector<pair<ll, ll>> querys(Q); rep(i, Q) { ll type, v = 0; cin >> type; if (type == 1)cin >> v; querys[i] = { type,v }; } // offline query's compression set<ll> st; for (const auto& query : querys) if (query.first == 1) st.insert(query.second); map<ll, ll> trans, rev; ll cnt = 0; for (auto& num : st) { trans[num] = cnt; rev[cnt] = num; cnt++; } // segment tree init auto f = [](ll a, ll b) {return a + b; }; SegmentTree<ll> seg(f, 0); seg.init(cnt); // query for (auto& query : querys) { ll type, v; tie(type, v) = query; if (type == 1) { ll pos = trans[v]; ll num = seg.query(pos, pos + 1); seg.set_val(pos, num + 1); } else { if (seg.query(0, cnt) < K) { //cerr << "debug"; cout << -1 << endl; continue; } // binary search ll out = -1, safe = cnt - 1; while (abs(safe - out) > 1) { ll mid = (out + safe) / 2; (seg.query(0, mid + 1) >= K ? safe : out) = mid; } //cerr << "debug"; cout << rev[safe] << endl; ll num = seg.query(safe, safe + 1); assert(num > 0); seg.set_val(safe, num - 1); } } return 0; }