結果

問題 No.59 鉄道の旅
コンテスト
ユーザー Kude
提出日時 2020-09-07 06:46:11
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 38 ms / 5,000 ms
コード長 996 bytes
コンパイル時間 1,357 ms
コンパイル使用メモリ 167,488 KB
実行使用メモリ 7,256 KB
最終ジャッジ日時 2024-11-29 10:12:18
合計ジャッジ時間 2,079 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

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