結果

問題 No.59 鉄道の旅
ユーザー Twizz
提出日時 2017-05-26 22:38:43
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 40 ms / 5,000 ms
コード長 852 bytes
コンパイル時間 408 ms
コンパイル使用メモリ 59,992 KB
実行使用メモリ 7,572 KB
最終ジャッジ日時 2024-09-19 20:13:02
合計ジャッジ時間 1,196 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
using namespace std;
typedef long long ll;
#define rep(i,n) for(int i=0;i<(n);i++)
//[1,n]
const int MAX_W = 1000010;
int bit[MAX_W + 1];

int sum(int i){//iまでの和を求める
    int s = 0;
    while(i > 0){
        s += bit[i];
        i -= i & -i;
    }
    return s;
}

void add(int i, int x){//iの値にxを加える
    while(i <= MAX_W){
        bit[i] += x;
        i += i & -i;
    }
}

int main(void){
    int n, k; cin >> n >> k;
    vector<int> w(n);
    rep(i, MAX_W + 1) bit[i] = 0;
    rep(i, n){
        int w; cin >> w;
        if(w > 0){
            if(sum(MAX_W) - sum(w - 1) < k){
                add(w, 1);
            }
        }else{
            if(sum(-w) - sum(-w - 1) >= 1){
                add(-w, -1);
            }
        }
    }
    cout << sum(MAX_W) << endl;
    return 0;
}
0