結果

問題 No.749 クエリ全部盛り
ユーザー risujiroh
提出日時 2018-11-08 00:39:20
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 974 ms / 3,000 ms
コード長 3,842 bytes
コンパイル時間 1,689 ms
コンパイル使用メモリ 185,120 KB
実行使用メモリ 143,904 KB
最終ジャッジ日時 2024-11-20 20:54:55
合計ジャッジ時間 8,100 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using lint = long long int;
template<class T = int> using V = vector<T>;
template<class T = int> using VV = V< V<T> >;
template<class T> void assign(V<T>& v, int n, const T& a = T()) { v.assign(n, a); }
template<class T, class... Args> void assign(V<T>& v, int n, const Args&... args) { v.resize(n); for (auto&& e : v) assign(e, args...); }


constexpr lint mod = 1e9 + 7;
inline lint emod(lint a, lint p = mod) { return (a % p + p) % p; }

struct Monoid {
  using T = struct { lint v, fib; };
  static T op(const T& lhs, const T& rhs) { return {(lhs.v + rhs.v) % mod, (lhs.fib + rhs.fib) % mod}; }
  static constexpr T e() { return {0, 0}; }
};

template<class Monoid> struct MonoidAct {
  using T = struct { lint a, b, c; };
  static void ap(const T& f, typename Monoid::T& a, int len) { a.v = (f.a * a.v + f.b * len + f.c * a.fib) % mod; }
  static void cp(const T& f, T& g) { (g.a *= f.a) %= mod, g.b = (f.a * g.b + f.b) % mod, g.c = (f.a * g.c + f.c) % mod; }
  static constexpr T id() { return {1, 0, 0}; }
};

template<class Monoid, class MonoidAct, class Int = int> struct SegmentTree {
  using Tv = typename Monoid::T;
  using Ta = typename MonoidAct::T;
  struct Node {
    Tv val = Monoid::e();
    Ta act = MonoidAct::id();
    unique_ptr<Node> cl, cr;
  };

  const Int n;
  unique_ptr<Node> root;

  SegmentTree(Int n) : n(n) {}

  Tv get(Int l, Int r) { return get(l, r, root, 0, n); }

  void act(Int l, Int r, const Ta& f) { act(l, r, f, root, 0, n); }

  void set(Int i, const Tv& a) { set(i, a, root, 0, n); }

  void push(const unique_ptr<Node>& v, Int vl, Int vm, Int vr) {
    if (!v->cl) v->cl = make_unique<Node>();
    MonoidAct::ap(v->act, v->cl->val, vm - vl);
    MonoidAct::cp(v->act, v->cl->act);
    if (!v->cr) v->cr = make_unique<Node>();
    MonoidAct::ap(v->act, v->cr->val, vr - vm);
    MonoidAct::cp(v->act, v->cr->act);
    v->act = MonoidAct::id();
  }

  Tv get(Int l, Int r, const unique_ptr<Node>& v, Int vl, Int vr) {
    if (!v) return Monoid::e();
    if (l <= vl and vr <= r) return v->val;
    Int vm = vl + vr >> 1;
    push(v, vl, vm, vr);
    Tv resl = l < vm ? get(l, r, v->cl, vl, vm) : Monoid::e(),
       resr = vm < r ? get(l, r, v->cr, vm, vr) : Monoid::e();
    return Monoid::op(resl, resr);
  }

  void act(Int l, Int r, const Ta& f, unique_ptr<Node>& v, Int vl, Int vr) {
    if (!v) v = make_unique<Node>();
    if (l <= vl and vr <= r) {
      MonoidAct::ap(f, v->val, vr - vl);
      MonoidAct::cp(f, v->act);
      return;
    }
    Int vm = vl + vr >> 1;
    push(v, vl, vm, vr);
    if (l < vm) act(l, r, f, v->cl, vl, vm);
    if (vm < r) act(l, r, f, v->cr, vm, vr);
    v->val = Monoid::op(v->cl ? v->cl->val : Monoid::e(), v->cr ? v->cr->val : Monoid::e());
  }

  void set(Int i, const Tv& a, unique_ptr<Node>& v, Int vl, Int vr) {
    if (!v) v = make_unique<Node>();
    if (vr - vl == 1) {
      v->val = a;
      return;
    }
    Int vm = vl + vr >> 1;
    push(v, vl, vm, vr);
    if (i < vm) set(i, a, v->cl, vl, vm);
    else set(i, a, v->cr, vm, vr);
    v->val = Monoid::op(v->cl ? v->cl->val : Monoid::e(), v->cr ? v->cr->val : Monoid::e());
  }
};

int main() {
  cin.tie(nullptr); ios_base::sync_with_stdio(false);
  int n, Q; cin >> n >> Q;
  V<Monoid::T> v(n);
  v[1].fib = 1;
  for (int i = 2; i < n; ++i) v[i].fib = (v[i - 1].fib + v[i - 2].fib) % mod;
  SegmentTree< Monoid, MonoidAct<Monoid> > st(n);
  for (int i = 0; i < n; ++i) st.set(i, v[i]);
  for (int i = 0; i < Q; ++i) {
    int q, l, r, k; cin >> q >> l >> r >> k, ++r;
    switch (q) {
      case 0: cout << emod(k * st.get(l, r).v) << '\n'; break;
      case 1: st.act(l, r, {0, k, 0}); break;
      case 2: st.act(l, r, {1, k, 0}); break;
      case 3: st.act(l, r, {k, 0, 0}); break;
      case 4: st.act(l, r, {1, 0, k}); break;
    }
  }
}
0