結果

問題 No.59 鉄道の旅
ユーザー ふーらくたるふーらくたる
提出日時 2016-07-05 01:18:12
言語 C++11
(gcc 13.3.0)
結果
AC  
実行時間 20 ms / 5,000 ms
コード長 1,443 bytes
コンパイル時間 635 ms
コンパイル使用メモリ 61,048 KB
実行使用メモリ 7,552 KB
最終ジャッジ日時 2024-12-24 22:22:30
合計ジャッジ時間 1,634 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
using namespace std;
/*
 * Fenwick Treeの実装.
 * 区間[1, n]において, いわゆるfoldlの操作を高速化することができる.
 */
template<typename T>
class FenwickTree {
public:
    vector<T> elt_;
    int n_;

    FenwickTree() { }

    FenwickTree(int n) {
        elt_ = vector<T>(n + 1);
        n_ = n;
        /* 初期化処理 */
        for (int i = 0; i <= n_; i++) {
            elt_[i] = 0;
        } 
    }

    void add(int i, T x) {
        while (i <= elt_.size()) {
            elt_[i] += x;
            i += i & -i;
        }
    }

    T query(int i) {
        T s = 0;
        while (i > 0) {
            /* クエリの処理 */
            s += elt_[i];
            i -= i & -i;
        }
        return s;
    }
};

const int kMAX_N = 100010;
const int kMAX_W = 1000010;

int N, K;
int W[kMAX_N];

void Solve() {
    FenwickTree<int> tree = FenwickTree<int>(kMAX_W);

    for (int i = 0; i < N; i++) {
        if (W[i] > 0 && tree.query(kMAX_W) - tree.query(W[i] - 1) < K) {
            tree.add(W[i], 1);
        } else if (W[i] < 0 && tree.query(abs(W[i])) - tree.query(abs(W[i]) - 1) > 0) {
            tree.add(abs(W[i]), -1);
        }
    }
    cout << tree.query(kMAX_W) << endl;
}

int main() {
    cin.tie(0);
    ios::sync_with_stdio(false);

    cin >> N >> K;
    for (int i = 0; i < N; i++) {
        cin >> W[i];
    }

    Solve();

    return 0;
}
0