結果
問題 | No.59 鉄道の旅 |
ユーザー | ninoinui |
提出日時 | 2020-09-06 03:02:15 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 38 ms / 5,000 ms |
コード長 | 1,013 bytes |
コンパイル時間 | 1,794 ms |
コンパイル使用メモリ | 195,724 KB |
最終ジャッジ日時 | 2025-01-14 07:21:18 |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 12 |
ソースコード
#include <bits/stdc++.h> using namespace std; template<typename T> class FenwickTree { int N, M; vector<T> B; public: FenwickTree(int n) : N(n + 1), M(1), B(N, 0) { while (M * 2 <= N) M *= 2; } FenwickTree(const vector<T> &V) : N((int) V.size() + 1), M(1), B(N, 0) { for (int i = 0; i < N - 1; i++) add(i, V.at(i)); while (M * 2 <= N) M *= 2; } void add(int i, T x) { i++; while (i < N) B.at(i) += x, i += i & -i; } void add(int l, int r, T x) { add(l, x); add(r + 1 , -x); } T sum(int i) { i++; T ret = 0; while (i > 0) ret += B.at(i), i -= i & -i; return ret; } T sum(int l, int r) { return sum(r) - sum(l - 1); } }; int main() { int N, K; cin >> N >> K; FenwickTree<long> FT(1e6+1); for (int i = 0; i < N; i++) { int W; cin >> W; if (W > 0) { if (FT.sum(W, 1e6) >= K) continue; FT.add(W, 1); } else { W = -W; if (FT.sum(W, W)) FT.add(W, -1); } } cout << FT.sum(1e6) << "\n"; }