結果

問題 No.59 鉄道の旅
コンテスト
ユーザー kroton
提出日時 2014-11-07 05:30:47
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 40 ms / 5,000 ms
コード長 811 bytes
コンパイル時間 459 ms
コンパイル使用メモリ 55,004 KB
実行使用メモリ 7,424 KB
最終ジャッジ日時 2024-12-24 20:21:18
合計ジャッジ時間 1,242 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
using namespace std;

const int MAX = 1000000;
int bit[MAX + 10];

int sum(int i){
    int s = 0;
    while(i > 0){
        s += bit[i];
        i -= i & -i;
    }
    return s;
}
void add(int i, int x){
    while(i <= MAX){
        bit[i] += x;
        i += i & -i;
    }
}

int main(){
    int N, K;
    cin >> N >> K;
    for(int i=0;i<N;i++){
        int W;
        cin >> W;
        
        if(W > 0){
            int cnt = sum(MAX) - sum(W - 1);
            if(cnt < K){
                add(W, 1);
            }
        } else {
            W = -W;
            
            int cnt = sum(W) - sum(W - 1);
            if(cnt >= 1){
                add(W, -1);
            }
        }
    }
    
    cout << sum(MAX) << endl;
    return 0;
}
0