結果

問題 No.1548 [Cherry 2nd Tune B] 貴方と私とサイクルとモーメント
ユーザー hitonanodehitonanode
提出日時 2021-06-11 23:29:55
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 103 ms / 4,500 ms
コード長 2,532 bytes
コンパイル時間 1,277 ms
コンパイル使用メモリ 110,788 KB
実行使用メモリ 20,300 KB
最終ジャッジ日時 2024-05-08 19:53:22
合計ジャッジ時間 6,252 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 71 ms
11,600 KB
testcase_03 AC 23 ms
7,612 KB
testcase_04 AC 70 ms
19,420 KB
testcase_05 AC 7 ms
5,376 KB
testcase_06 AC 51 ms
10,996 KB
testcase_07 AC 64 ms
20,072 KB
testcase_08 AC 69 ms
11,768 KB
testcase_09 AC 49 ms
7,656 KB
testcase_10 AC 54 ms
19,116 KB
testcase_11 AC 62 ms
19,156 KB
testcase_12 AC 34 ms
5,384 KB
testcase_13 AC 42 ms
19,136 KB
testcase_14 AC 51 ms
7,176 KB
testcase_15 AC 43 ms
20,208 KB
testcase_16 AC 59 ms
12,076 KB
testcase_17 AC 78 ms
18,960 KB
testcase_18 AC 68 ms
19,892 KB
testcase_19 AC 28 ms
7,192 KB
testcase_20 AC 49 ms
19,440 KB
testcase_21 AC 46 ms
11,152 KB
testcase_22 AC 97 ms
20,208 KB
testcase_23 AC 100 ms
20,216 KB
testcase_24 AC 103 ms
20,220 KB
testcase_25 AC 95 ms
20,300 KB
testcase_26 AC 96 ms
20,220 KB
testcase_27 AC 98 ms
20,212 KB
testcase_28 AC 103 ms
20,088 KB
testcase_29 AC 96 ms
20,216 KB
testcase_30 AC 98 ms
20,216 KB
testcase_31 AC 95 ms
20,216 KB
testcase_32 AC 75 ms
20,216 KB
testcase_33 AC 85 ms
20,216 KB
testcase_34 AC 79 ms
20,216 KB
testcase_35 AC 79 ms
20,240 KB
testcase_36 AC 81 ms
20,220 KB
testcase_37 AC 96 ms
20,212 KB
testcase_38 AC 76 ms
20,212 KB
testcase_39 AC 83 ms
20,084 KB
testcase_40 AC 80 ms
20,212 KB
testcase_41 AC 81 ms
20,180 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <utility>
#include <vector>
using namespace std;

#include <atcoder/modint>
#include <atcoder/lazysegtree>
using mint = atcoder::modint998244353;
struct S {
    mint sum0, sum1, sum2, sum3, sum4;
};
S op(S l, S r) {
    return {l.sum0 + r.sum0, l.sum1 + r.sum1, l.sum2 + r.sum2, l.sum3 + r.sum3, l.sum4 + r.sum4};
}
S e() { return {0, 0, 0, 0, 0}; }
using F = pair<bool, mint>;
S mp(F f, S x) {
    if (!f.first) return x;
    auto &v = f.second;
    return {x.sum0, x.sum0 * v, x.sum0 * v * v, x.sum0 * v * v * v, x.sum0 * v * v * v * v};
}
F composition(F fnew, F gold) {
    if (fnew.first) return fnew;
    return gold;
}
F id() { return {false, 0}; }

template <int md> std::istream &operator>>(std::istream &is, atcoder::static_modint<md> &x) {
    long long t;
    return is >> t, x = t, is;
}
template <int md> std::ostream &operator<<(std::ostream &os, const atcoder::static_modint<md> &x) {
    return os << x.val();
}


int main() {
    cin.tie(nullptr), ios::sync_with_stdio(false);
    int N;
    cin >> N;
    vector<mint> A(N);
    for (auto &x : A) cin >> x;
    vector<S> init;
    for (auto x : A) {
        init.push_back({1, x, x * x, x * x * x, x * x * x * x});
    }
    atcoder::lazy_segtree<S, op, e, F, mp, composition, id> tree(init);
    int Q;
    cin >> Q;
    while (Q--) {
        int type;
        int u, v, w;
        cin >> type >> u >> v >> w;
        vector<pair<int, int>> ranges;
        if (u > v) swap(u, v);
        if (u < w and w < v) {
            ranges.emplace_back(u - 1, v);
        } else {
            ranges.emplace_back(0, u), ranges.emplace_back(v - 1, N);
        }
        if (type == 0) {
            mint b;
            cin >> b;
            for (auto [l, r] : ranges) tree.apply(l, r, {true, b});
        } else {
            auto ret = e();
            for (auto [l, r] : ranges) ret = op(ret, tree.prod(l, r));
            const mint mu = ret.sum1 / ret.sum0;
            if (type == 1) {
                cout << "0\n";
            } else if (type == 2) {
                cout << (ret.sum2 - 2 * ret.sum1 * mu + ret.sum0 * mu * mu) / ret.sum0 << '\n';
            } else if (type == 3) {
                cout << (ret.sum3 - 3 * ret.sum2 * mu + 3 * ret.sum1 * mu * mu - ret.sum0 * mu * mu * mu) / ret.sum0 << '\n';
            } else if (type == 4) {
                cout << (ret.sum4 - 4 * ret.sum3 * mu + 6 * ret.sum2 * mu * mu - 4 * ret.sum1 * mu * mu * mu + ret.sum0 * mu * mu * mu * mu) / ret.sum0 << '\n';
            }
        }
    }
}
0