#132600 (C++14) No.59 鉄道の旅

• 問題: No.59 鉄道の旅
• ユーザー: izuru_matsuura
• 提出日時: 2016-11-22 18:24:10
• 言語: C++14 (gcc 13.3.0 + boost 1.87.0)
• 結果: AC
• 実行時間: 157 ms / 5,000 ms
• コード長: 2,731 bytes
• コンパイル時間: 2,177 ms
• コンパイル使用メモリ: 178,704 KB
• 実行使用メモリ: 40,832 KB
• 最終ジャッジ日時: 2024-12-25 00:02:55
• 合計ジャッジ時間: 3,820 ms
• ジャッジサーバーID （参考情報）: judge2 / judge1

ファイルパターン	結果
sample	AC * 4
other	AC * 12

ソースコード

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
#include <bits/stdc++.h>
using namespace std;
namespace {
    typedef double real;
    typedef long long ll;
    template<class T> ostream& operator<<(ostream& os, const vector<T>& vs) {
        if (vs.empty()) return os << "[]";
        auto i = vs.begin();
        os << "[" << *i;
        for (++i; i != vs.end(); ++i) os << " " << *i;
        return os << "]";
    }
    template<class T> istream& operator>>(istream& is, vector<T>& vs) {
        for (auto it = vs.begin(); it != vs.end(); it++) is >> *it;
        return is;
    }
    int N, K;
    void input() {
        cin >> N >> K;
    }
    typedef long long Long;
    /* 蛹ｺ髢薙∈縺ｮ荳·讒伜刈邂励↓蟇ｾ蠢懊＠縺欒angeSumQuery */
    struct RangeSumQuery2 {
        // root縺ｮindex縺ｯ1
        int N;
        vector<Long> V, L;
        static int FindPow2(int x) {
            int x1 = 1;
            while (x1 < x) x1 *= 2;
            return x1;
        }
        RangeSumQuery2(int N_) : N(FindPow2(N_)) {
            V = vector<Long>(N * 2, 0);
            L = vector<Long>(N * 2, 0);
        }
        /* [s, t)縺ｫ荳·讒倥↓x繧貞刈邂・*/
        void add(Long x, int s, int t) { return add(x, s, t, 0, N, 1); }
        void add(Long x, int s, int t, int a, int b, int k) {
            if (t <= a || b <= s) return;
            if (s <= a && b <= t) {
                L[k] += x;
                return;
            }
            if (s < b || a < t) {
                V[k] += (min(t, b) - max(s, a)) * x;
                if (b - a == 1) return;
                int m = (a + b) / 2;
                add(x, s, t, a, m, k * 2);
                add(x, s, t, m, b, k * 2 + 1);
            }
        }
        /* [s, t)縺ｮ蜥・*/
        Long query(int s, int t) { return query(s, t, 0, N, 1); }
        Long query(int s, int t, int a, int b, int k) {
            if (t <= a || b <= s) return 0;
            if (s <= a && b <= t) return V[k] + L[k] * (b - a);
            int m = (a + b) / 2;
            return (min(t, b) - max(s, a)) * L[k] +
                query(s, t, a, m, k * 2) +
                query(s, t, m, b, k * 2 + 1);
        }
    };
    void solve() {
        RangeSumQuery2 rsq(1000000 + 1);
        map<int, int> m;
        for (int i = 0; i < N; i++) {
            int W; cin >> W;
            if (W < 0) {
                W = -W;
                if (m[W] == 0) continue;
                rsq.add(-1, 0, W + 1);
                m[W]--;
            } else {
                if (rsq.query(W, W + 1) >= K) continue;
                m[W]++;
                rsq.add(1, 0, W + 1);
            }
        }
        cout << rsq.query(0, 1) << endl;
    }
}
int main() {
    input(); solve();
    return 0;
}
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX