結果

問題 No.1548 [Cherry 2nd Tune B] 貴方と私とサイクルとモーメント
ユーザー hitonanode
提出日時 2021-06-11 23:29:55
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 210 ms / 4,500 ms
コード長 2,532 bytes
コンパイル時間 3,241 ms
コンパイル使用メモリ 108,640 KB
最終ジャッジ日時 2025-01-22 07:06:08
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 42
権限があれば一括ダウンロードができます

ソースコード

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