結果

問題 No.749 クエリ全部盛り
ユーザー magurofly
提出日時 2021-04-22 03:04:59
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 505 ms / 3,000 ms
コード長 1,821 bytes
コンパイル時間 2,280 ms
コンパイル使用メモリ 204,344 KB
最終ジャッジ日時 2025-01-20 22:33:49
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#include "atcoder/lazysegtree"
using namespace std;
using namespace atcoder;

using ll = long long;

const ll MOD = 1000000007;

typedef struct {
    ll len;
    ll sum;
    ll fib;
} S;
S op(S x, S y) {
    return { (x.len + y.len) % MOD, (x.sum + y.sum) % MOD, (x.fib + y.fib) % MOD };
}
S e() {
    return { 0, 0, 0 };
}

typedef struct {
    ll add;
    ll fib;
    ll mul;
    bool updates;
} F;
S mapping(F f, S x) {
    if (f.updates) {
        return { x.len, (x.len * f.add + x.fib * f.fib) % MOD, x.fib };
    } else {
        return { x.len, (x.len * f.add + x.sum * f.mul + x.fib * f.fib) % MOD, x.fib };
    }
}
F composition(F f, F g) {
    if (f.updates) {
        return f;
    } else {
        return { (f.mul * g.add + f.add) % MOD, (f.mul * g.fib + f.fib) % MOD, (f.mul * g.mul) % MOD, g.updates };
    }
}
F id() {
    return { 0, 0, 1, false };
}

int main() {
    int N, Q; cin >> N >> Q;
    vector<S> a(N, { 1, 0, 0 });
    a[1].fib = 1;
    for(int i = 2; i < N; i++) a[i].fib = (a[i - 2].fib + a[i - 1].fib) % MOD;
    lazy_segtree<S, op, e, F, mapping, composition, id> seg(a);
    for(int i = Q; i--; ) {
        int q, l, r, k; cin >> q >> l >> r >> k; r++;
        if (q == 0) {
            S x = seg.prod(l, r);
            cout << k * x.sum % MOD << endl;
        } else if (q == 1) {
            seg.apply(l, r, { k, 0, 1, true });
        } else if (q == 2) {
            seg.apply(l, r, { k, 0, 1, false });
        } else if (q == 3) {
            seg.apply(l, r, { 0, 0, k, false });
        } else if (q == 4) {
            seg.apply(l, r, { 0, k, 1, false });
        }

        // cerr << "cur:";
        // for (int j = 0; j < N; j++) {
        //   S x = seg.get(j);
        //   cerr << ' ' << x.sum % MOD;
        // }
        // cerr << endl;
    }

}
0