結果

問題 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,586 ms
コンパイル使用メモリ 110,868 KB
実行使用メモリ 21,312 KB
最終ジャッジ日時 2024-12-15 02:54:36
合計ジャッジ時間 6,707 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,816 KB
testcase_02 AC 73 ms
11,568 KB
testcase_03 AC 23 ms
7,520 KB
testcase_04 AC 69 ms
20,760 KB
testcase_05 AC 8 ms
6,820 KB
testcase_06 AC 54 ms
11,032 KB
testcase_07 AC 63 ms
21,312 KB
testcase_08 AC 76 ms
11,848 KB
testcase_09 AC 52 ms
7,768 KB
testcase_10 AC 51 ms
19,328 KB
testcase_11 AC 59 ms
20,236 KB
testcase_12 AC 36 ms
6,816 KB
testcase_13 AC 38 ms
19,160 KB
testcase_14 AC 54 ms
7,296 KB
testcase_15 AC 40 ms
20,328 KB
testcase_16 AC 60 ms
12,124 KB
testcase_17 AC 79 ms
19,812 KB
testcase_18 AC 66 ms
20,220 KB
testcase_19 AC 30 ms
7,316 KB
testcase_20 AC 48 ms
19,612 KB
testcase_21 AC 50 ms
11,064 KB
testcase_22 AC 96 ms
20,860 KB
testcase_23 AC 94 ms
20,200 KB
testcase_24 AC 99 ms
20,208 KB
testcase_25 AC 103 ms
20,168 KB
testcase_26 AC 97 ms
20,332 KB
testcase_27 AC 97 ms
20,892 KB
testcase_28 AC 96 ms
20,164 KB
testcase_29 AC 94 ms
20,380 KB
testcase_30 AC 100 ms
20,204 KB
testcase_31 AC 97 ms
20,284 KB
testcase_32 AC 75 ms
21,236 KB
testcase_33 AC 85 ms
20,212 KB
testcase_34 AC 80 ms
20,196 KB
testcase_35 AC 81 ms
20,740 KB
testcase_36 AC 86 ms
20,164 KB
testcase_37 AC 96 ms
20,044 KB
testcase_38 AC 77 ms
20,712 KB
testcase_39 AC 85 ms
20,196 KB
testcase_40 AC 80 ms
20,196 KB
testcase_41 AC 82 ms
20,436 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