結果

問題 No.59 鉄道の旅
コンテスト
ユーザー tottoripaper
提出日時 2014-12-08 21:57:17
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 54 ms / 5,000 ms
コード長 1,393 bytes
コンパイル時間 203 ms
コンパイル使用メモリ 26,584 KB
実行使用メモリ 9,728 KB
最終ジャッジ日時 2024-12-24 20:28:32
合計ジャッジ時間 1,095 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 12
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:48:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   48 |     scanf("%d %d", &N, &K);
      |     ~~~~~^~~~~~~~~~~~~~~~~
main.cpp:52:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   52 |         scanf("%d", &w);
      |         ~~~~~^~~~~~~~~~

ソースコード

diff # 

#include <cstdio>

#define FOR(i,a,b) for(int i=(a);i<(b);i++)
#define REP(i,j) FOR(i,0,j)

const int MAX_N = 1000001;

class SegmentTree{
public:
    SegmentTree(){}
    SegmentTree(int n){init(n);}
    void init(int n){
        size = 1;
        while(size < n){
            size *= 2;
        }
        
        REP(i, size*2-1){
            data[i] = 0;
        }
    }
    void update(int k, int v){
        k += size - 1;
        data[k] = v;
        
        while(k > 0){
            k = (k-1) / 2;
            data[k] = data[k*2+1] + data[k*2+2];
        }
    }
    int query(int l, int r){
        return _query(l, r, 0, 0, size);
    }
private:
    int _query(int a, int b, int k, int l, int r){
        if(b <= l || r <= a){return 0;}
        if(a <= l && r <= b){return data[k];}
        return _query(a, b, 2*k+1, l, (l+r)/2) +
            _query(a, b, 2*k+2, (l+r)/2, r);
    }
    int data[MAX_N*4], size;
};

SegmentTree st(MAX_N);

int main(){
    int N, K;
    scanf("%d %d", &N, &K);

    for(int i=0;i<N;i++){
        int w;
        scanf("%d", &w);

        if(w < 0 && 0 < st.query(-w, -w+1)){
            st.update(-w, st.query(-w, -w+1) - 1);
        }else if(w > 0 && st.query(w, 1000001) < K){
            st.update(w, st.query(w, w+1) + 1);
        }
    }

    printf("%d\n", st.query(0, 1000001));
}
0