結果

問題 No.59 鉄道の旅
ユーザー KudeKude
提出日時 2020-09-07 06:46:11
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 39 ms / 5,000 ms
コード長 996 bytes
コンパイル時間 1,421 ms
コンパイル使用メモリ 168,252 KB
実行使用メモリ 7,176 KB
最終ジャッジ日時 2024-05-06 23:21:18
合計ジャッジ時間 2,164 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
7,088 KB
testcase_01 AC 3 ms
7,164 KB
testcase_02 AC 4 ms
7,104 KB
testcase_03 AC 4 ms
7,168 KB
testcase_04 AC 36 ms
7,172 KB
testcase_05 AC 4 ms
7,168 KB
testcase_06 AC 3 ms
7,060 KB
testcase_07 AC 3 ms
7,120 KB
testcase_08 AC 7 ms
7,088 KB
testcase_09 AC 8 ms
7,168 KB
testcase_10 AC 8 ms
7,128 KB
testcase_11 AC 5 ms
7,124 KB
testcase_12 AC 19 ms
7,108 KB
testcase_13 AC 39 ms
7,176 KB
testcase_14 AC 37 ms
7,160 KB
testcase_15 AC 4 ms
7,160 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#define rep(i,n) for (int i = 0; i < (n); ++i)
#define chmax(a, b) a = max(a, b)
#define chmin(a, b) a = min(a, b)
#define all(x) (x).begin(), (x).end()
using namespace std;
using ll = long long;
using P = pair<int,int>;
using VI = vector<int>;
using VVI = vector<VI>;
template<typename T>
struct BIT {
  int n;
  vector<T> d;
  BIT(int n=0):n(n),d(n+1) {}
  void add(int i, T x=1) {
    for (i++; i <= n; i += i&-i) {
      d[i] += x;
    }
  }
  T sum(int i) {
    T x = 0;
    for (i++; i; i -= i&-i) {
      x += d[i];
    }
    return x;
  }
  T sum(int l, int r) {
    return sum(r-1) - sum(l-1);
  }
  int operator[](int x) {
      return sum(x, x + 1);
  }
};
int main() {
    int n, k;
    cin >> n >> k;
    BIT<int> d(1000001);
    rep(_, n) {
        int w;
        cin >> w;
        if (w > 0) {
            if (d.sum(w, 1000001) < k) d.add(w);
        } else {
            if (d[-w] > 0) d.add(-w, -1);
        }
    }
    cout << d.sum(1000000) << endl;
}
0