結果

問題 No.255 Splarrraaay スプラーレェーーイ
ユーザー maine_honzukimaine_honzuki
提出日時 2021-05-04 22:37:26
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,139 ms / 10,000 ms
コード長 2,478 bytes
コンパイル時間 4,788 ms
コンパイル使用メモリ 275,108 KB
実行使用メモリ 160,932 KB
最終ジャッジ日時 2023-10-01 00:17:14
合計ジャッジ時間 17,322 ms
ジャッジサーバーID
(参考情報)
judge15 / judge12
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1,139 ms
160,736 KB
testcase_01 AC 1,127 ms
160,596 KB
testcase_02 AC 1,061 ms
160,676 KB
testcase_03 AC 1,110 ms
160,856 KB
testcase_04 AC 1 ms
4,380 KB
testcase_05 AC 1,064 ms
160,552 KB
testcase_06 AC 1,112 ms
160,892 KB
testcase_07 AC 1,108 ms
160,932 KB
testcase_08 AC 1,079 ms
160,592 KB
testcase_09 AC 1,059 ms
160,480 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

//https://ncode.syosetu.com/n4830bu/255/
#include "atcoder/all"
#include <bits/stdc++.h>
using namespace atcoder;
using namespace std;

using ll = long long;
const ll mod = 1000000000000000009;

struct S {
    ll maine, len;
};
S op(S a, S b) {
    return {a.maine + b.maine, a.len + b.len};
}
S e() {
    return {0, 0};
}
struct F {
    ll book;
    bool zero;
};
S mapping(F f, S x) {
    if (f.zero)
        x.maine = 0;
    x.maine += f.book * x.len;
    return x;
}
F composition(F f, F g) {
    if (f.zero)
        return f;
    if (g.zero)
        f.zero = true;
    f.book += g.book;
    return f;
}
F id() {
    return {0, false};
}

int main() {
    ll N;
    int Q;
    cin >> N >> Q;
    vector<int> X(Q);
    vector<ll> L(Q), R(Q);
    for (int i = 0; i < Q; i++) {
        cin >> X[i] >> L[i] >> R[i];
        X[i]--;
        R[i]++;
    }
    vector<ll> compress{0, N};
    compress.insert(compress.end(), L.begin(), L.end());
    compress.insert(compress.end(), R.begin(), R.end());
    sort(compress.begin(), compress.end());
    compress.erase(unique(begin(compress), end(compress)), end(compress));
    auto idx = [&](ll x) { return lower_bound(compress.begin(), compress.end(), x) - compress.begin(); };
    int M = compress.size();
    vector<S> initial(M - 1);
    for (int i = 0; i < M - 1; i++) {
        initial[i] = {0, compress[i + 1] - compress[i]};
    }
    vector segs(5, lazy_segtree<S, op, e, F, mapping, composition, id>(initial));
    vector<ll> bonus(5);
    for (int i = 0; i < Q; i++) {
        int x = X[i];
        ll l = L[i], r = R[i];
        int l_idx = idx(l), r_idx = idx(r);
        if (x >= 0) {
            for (int y = 0; y < 5; y++) {
                if (x == y)
                    continue;
                segs[y].apply(l_idx, r_idx, {0, true});
            }
            segs[x].apply(l_idx, r_idx, {1, false});
        } else {
            vector<ll> scores(5);
            for (int x = 0; x < 5; x++) {
                scores[x] = segs[x].prod(l_idx, r_idx).maine;
            }
            int idx_max = max_element(scores.begin(), scores.end()) - scores.begin();
            int cnt = 0;
            for (int x = 0; x < 5; x++) {
                cnt += scores[x] == scores[idx_max];
            }
            if (cnt == 1)
                (bonus[idx_max] += scores[idx_max]) %= mod;
        }
    }
    for (int x = 0; x < 5; x++) {
        cout << (segs[x].all_prod().maine + bonus[x]) % mod << " \n"[x == 4];
    }
}
0