結果

問題 No.59 鉄道の旅
ユーザー codershifthcodershifth
提出日時 2015-07-27 23:54:58
言語 C++11
(gcc 11.4.0)
結果
RE  
実行時間 -
コード長 1,994 bytes
コンパイル時間 1,396 ms
コンパイル使用メモリ 161,908 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-06-07 00:57:20
合計ジャッジ時間 3,826 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
5,248 KB
testcase_01 AC 3 ms
5,376 KB
testcase_02 AC 3 ms
5,376 KB
testcase_03 AC 3 ms
5,376 KB
testcase_04 RE -
testcase_05 RE -
testcase_06 RE -
testcase_07 RE -
testcase_08 RE -
testcase_09 RE -
testcase_10 RE -
testcase_11 AC 3 ms
5,376 KB
testcase_12 AC 10 ms
5,376 KB
testcase_13 RE -
testcase_14 RE -
testcase_15 RE -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

typedef long long ll;
typedef unsigned long long ull;

#define FOR(i,a,b) for(int (i)=(a);i<(b);i++)
#define REP(i,n) FOR(i,0,n)
#define RANGE(vec) (vec).begin(),(vec).end()

using namespace std;

class RailroadTrip {
public:
    void solve(void) {
            int N,K;
            cin>>N>>K;
            // 単純にシミュレーションすればよい
            // 駅数が最大 10^5 なので積み荷の載せ・降ろしの判定を高速でできればよい。

            // 次のクエリを log(N) くらいでできればよい
            //  * a[w] += 1
            //  * a[w]+a[w+1]+...+a[maxW] を計算
            // BIT を使う

            int maxW = (1<<(int)ceil(log2(100000)));

            // [1,maxW]
            vector<ll> bits(maxW+1,0);
            // a[i] に x を追加する
            auto add = [&](int i, int x) {
                assert(i>0);
                while (i <= maxW)
                {
                    bits[i] += x;
                    i += (i & -i);
                }
            };
            // i 以上の a[i] の和
            auto sum = [&](int i) {
                ll s = 0;
                --i; // 求めたいのは (a[1]+...+a[maxW]) - (a[1]+...+a[i-1]) なので
                while (i > 0)
                {
                    s += bits[i];
                    i -= (i & -i);
                }
                return bits[maxW] - s;
            };
            // O(N*log(N))
            REP(i,N)
            {
                ll w, aw;
                cin>>w;
                aw = abs(w);
                if (w > 0 && sum(w) < K)
                    add(w,1);
                else if (w < 0 && sum(aw)-sum(aw+1) > 0)
                    add(aw,-1);
            }
            cout<<sum(1)<<endl;
    }
};

#if 1
int main(int argc, char *argv[])
{
        ios::sync_with_stdio(false);
        auto obj = new RailroadTrip();
        obj->solve();
        delete obj;
        return 0;
}
#endif
0