結果
問題 |
No.59 鉄道の旅
|
ユーザー |
![]() |
提出日時 | 2020-05-14 10:51:49 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 81 ms / 5,000 ms |
コード長 | 2,372 bytes |
コンパイル時間 | 1,814 ms |
コンパイル使用メモリ | 176,024 KB |
実行使用メモリ | 28,160 KB |
最終ジャッジ日時 | 2024-09-15 00:52:24 |
合計ジャッジ時間 | 3,385 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 12 |
ソースコード
#include <bits/stdc++.h> using namespace std; typedef long long ll; #define rep(i, n) for(ll i = 0, i##_len = (n); i < i##_len; i++) #define reps(i, s, n) for(ll i = (s), i##_len = (n); i < i##_len; i++) #define rrep(i, n) for(ll i = (n) - 1; i >= 0; i--) #define rreps(i, e, n) for(ll i = (n) - 1; i >= (e); i--) #define all(x) (x).begin(), (x).end() #define rall(x) (x).rbegin(), (x).rend() #define sz(x) ((ll)(x).size()) #define len(x) ((ll)(x).length()) #define endl "\n" template<class T> struct SegmentTree { private: int n; vector<T> node; T def; function<T(T,T)> lop; function<T(T,T)> out; public: SegmentTree() {} SegmentTree(const vector<T> &v, T _def, function<T(T,T)> _lop, function<T(T,T)> _out) { int vl = (int)v.size(); n = 1; while(n < vl) n *= 2; def = _def; lop = _lop; out = _out; node = vector<T>(2 * n - 1, def); for(int i = 0; i < vl; i++) node[i + n - 1] = v[i]; for(int i = n - 2; i >= 0; i--) node[i] = out(node[i * 2 + 1], node[i * 2 + 2]); } void update(int idx, T val) { idx += (n - 1); node[idx] = lop(node[idx], val); while(idx > 0) { idx = (idx - 1) / 2; node[idx] = out(node[idx * 2 + 1], node[idx * 2 + 2]); } } T query(int a, int b, int k = 0, int l = 0, int r = -1) { if (r < 0) r = n; if ((r <= a) || (b <= l)) return def; if ((a <= l) && (r <= b)) return node[k]; return out(query(a, b, 2 * k + 1, l, (l + r) / 2), query(a, b, 2 * k + 2, (l + r) / 2, r)); } }; int main() { cin.tie(0); ios::sync_with_stdio(false); // ifstream in("input.txt"); // cin.rdbuf(in.rdbuf()); ll n, k; cin >> n >> k; vector<ll> w(n); rep(i, n) cin >> w[i]; auto lop = [](ll a, ll b){ return a + b; }; auto out = [](ll a, ll b){ return a + b; }; SegmentTree<ll> st(vector<ll>(1000001, 0), 0, lop, out); rep(i, n) { if (w[i] < 0) { w[i] *= -1; ll v = st.query(w[i], w[i] + 1); if (v > 0) { st.update(w[i], -1); } } else { ll v = st.query(w[i], 1000001); if (v < k) { st.update(w[i], 1); } } } cout << st.query(0, 1000001) << endl; return 0; }