結果

問題 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,131 ms
コンパイル使用メモリ 95,956 KB
実行使用メモリ 21,260 KB
最終ジャッジ日時 2023-08-21 14:29:37
合計ジャッジ時間 6,368 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,380 KB
testcase_01 AC 1 ms
4,380 KB
testcase_02 AC 71 ms
11,480 KB
testcase_03 AC 23 ms
7,368 KB
testcase_04 AC 67 ms
19,288 KB
testcase_05 AC 8 ms
4,376 KB
testcase_06 AC 53 ms
10,840 KB
testcase_07 AC 63 ms
20,092 KB
testcase_08 AC 72 ms
11,456 KB
testcase_09 AC 52 ms
7,676 KB
testcase_10 AC 51 ms
19,300 KB
testcase_11 AC 60 ms
19,512 KB
testcase_12 AC 36 ms
5,264 KB
testcase_13 AC 39 ms
19,340 KB
testcase_14 AC 53 ms
7,104 KB
testcase_15 AC 43 ms
20,056 KB
testcase_16 AC 60 ms
12,072 KB
testcase_17 AC 78 ms
18,824 KB
testcase_18 AC 65 ms
20,120 KB
testcase_19 AC 29 ms
7,388 KB
testcase_20 AC 48 ms
19,976 KB
testcase_21 AC 48 ms
11,280 KB
testcase_22 AC 97 ms
21,128 KB
testcase_23 AC 96 ms
19,852 KB
testcase_24 AC 97 ms
19,924 KB
testcase_25 AC 102 ms
20,080 KB
testcase_26 AC 103 ms
20,052 KB
testcase_27 AC 100 ms
20,876 KB
testcase_28 AC 100 ms
20,248 KB
testcase_29 AC 99 ms
20,176 KB
testcase_30 AC 99 ms
21,080 KB
testcase_31 AC 98 ms
19,984 KB
testcase_32 AC 75 ms
20,096 KB
testcase_33 AC 86 ms
20,528 KB
testcase_34 AC 81 ms
19,968 KB
testcase_35 AC 80 ms
20,688 KB
testcase_36 AC 84 ms
19,916 KB
testcase_37 AC 95 ms
19,940 KB
testcase_38 AC 75 ms
20,880 KB
testcase_39 AC 85 ms
20,208 KB
testcase_40 AC 82 ms
21,260 KB
testcase_41 AC 81 ms
20,516 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