結果
問題 | No.59 鉄道の旅 |
ユーザー |
![]() |
提出日時 | 2020-05-08 23:13:12 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
RE
|
実行時間 | - |
コード長 | 1,031 bytes |
コンパイル時間 | 1,507 ms |
コンパイル使用メモリ | 167,776 KB |
実行使用メモリ | 6,948 KB |
最終ジャッジ日時 | 2024-07-04 01:12:20 |
合計ジャッジ時間 | 4,269 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 2 RE * 10 |
ソースコード
#include <bits/stdc++.h>using namespace std;template <typename T>struct BinaryIndexedTree {vector<T> data;int n;BinaryIndexedTree(int n_) : n(n_ + 1), data(n_ + 1, 0) {}T sum(int i) {T ret(0);for (int x = i; x > 0; x -= x & -x)ret += data[x];return ret;}void add(int i, T a) {for (int x = i; x <= n; x += x & -x)data[x] += a;}};int main() {int N, K, W[100010];cin >> N >> K;for (int i = 0; i < N; i++) {cin >> W[i];}int cnt[100020] = {};BinaryIndexedTree<int> bit(100020);int MAX = 100010;for (int i = 0; i < N; i++) {if (W[i] > 0) {if (bit.sum(MAX) - bit.sum(W[i] - 1) < K) {bit.add(W[i], 1);cnt[W[i]]++;}} else {W[i] *= -1;if (cnt[W[i]]) {cnt[W[i]]--;bit.add(W[i], -1);}}}cout << bit.sum(MAX) << endl;}