結果

問題 No.59 鉄道の旅
コンテスト
ユーザー 🍮かんプリン
提出日時 2019-09-09 15:52:17
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 42 ms / 5,000 ms
コード長 1,625 bytes
コンパイル時間 1,574 ms
コンパイル使用メモリ 169,820 KB
実行使用メモリ 7,168 KB
最終ジャッジ日時 2024-06-28 12:24:57
合計ジャッジ時間 2,585 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
#define ALL(obj) (obj).begin(),(obj).end()
#define RALL(obj) (obj).rbegin(),(obj).rend()
#define REP(i, n) for(int i = 0; i < (int)(n); i++)
#define REPR(i, n) for(int i = (int)(n); i >= 0; i--)
#define FOR(i,n,m) for(int i = (int)(n); i < int(m); i++)
using namespace std;
typedef long long ll;
const int MOD = 1e9 + 7;
const int INF = MOD - 1;
const ll LLINF = 4e18;

// 0-indexedとして使う
// 実装は1-indexed
template< typename T >
struct BinaryIndexedTree {
    vector< T > data;

    BinaryIndexedTree(int sz) {
        data.assign(++sz, 0);
    }

    // [0,k]
    T sum(int k) {
        T ret = 0;
        for (++k; k > 0; k -= k & -k) ret += data[k];
        return (ret);
    }

    // [left,right]
    T sum(int left, int right) {
        return sum(right) - sum(left - 1);
    }

    // 負の数も可
    void add(int k, T x) {
        for (++k; k < data.size(); k += k & -k) data[k] += x;
    }

    void print() {
        cout << "[ ";
        for (int i = 1; i < data.size(); i++) {
            cout << data[i];
            if (i < data.size() - 1) cout << ", ";
        }
        cout << " ]";
    }
};

int main() {
    int n, k; cin >> n >> k;
    const int max_w = 1000000;
    BinaryIndexedTree<int> bit(max_w+1);
    REP(i, n) {
        int w; cin >> w;
        if (w > 0) {
            if (bit.sum(w, max_w) < k) {
                bit.add(w, 1);
            }
        }
        else {
            w = -w;
            if (bit.sum(w,w) > 0) {
                bit.add(w, -1);
            }
        }
    }
    cout << bit.sum(0, max_w) << endl;
    getchar(); getchar();
}
0