結果
問題 | No.59 鉄道の旅 |
ユーザー | toma |
提出日時 | 2019-09-24 22:30:19 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 58 ms / 5,000 ms |
コード長 | 1,646 bytes |
コンパイル時間 | 1,655 ms |
コンパイル使用メモリ | 172,736 KB |
実行使用メモリ | 20,388 KB |
最終ジャッジ日時 | 2024-09-19 13:06:18 |
合計ジャッジ時間 | 2,841 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 7 ms
19,584 KB |
testcase_01 | AC | 8 ms
19,596 KB |
testcase_02 | AC | 8 ms
19,584 KB |
testcase_03 | AC | 8 ms
19,636 KB |
testcase_04 | AC | 58 ms
20,360 KB |
testcase_05 | AC | 8 ms
19,584 KB |
testcase_06 | AC | 8 ms
19,584 KB |
testcase_07 | AC | 9 ms
19,584 KB |
testcase_08 | AC | 15 ms
19,712 KB |
testcase_09 | AC | 14 ms
19,592 KB |
testcase_10 | AC | 14 ms
19,712 KB |
testcase_11 | AC | 10 ms
19,676 KB |
testcase_12 | AC | 31 ms
20,388 KB |
testcase_13 | AC | 53 ms
20,324 KB |
testcase_14 | AC | 58 ms
20,352 KB |
testcase_15 | AC | 7 ms
19,584 KB |
ソースコード
#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 N, K; cin >> N >> K; vector<ll> W(N); rep(i, N)cin >> W[i]; // preprocess constexpr ll lim = 1e6 + 10; auto f = [](ll a, ll b) {return a + b; }; SegmentTree<ll> seg(f, 0); seg.init(lim); // query for (auto w : W) { if (w > 0) { ll now = seg.query(w, w + 1); if (seg.query(w, lim) < K) { seg.set_val(w, now + 1); } } else { w *= -1; ll now = seg.query(w, w + 1); now = max(now - 1, 0ll); seg.set_val(w, now); } } cout << seg.query(0, lim) << endl; return 0; }