test/aoj/DSL_2_I.test.cpp

View this file on GitHub
Last update: 2021-09-04 15:28:55+09:00
Problem: https://onlinejudge.u-aizu.ac.jp/courses/library/3/DSL/all/DSL_2_I

Depends on

DataStructure/lazy_segment_tree.cpp

Code

#define PROBLEM "https://onlinejudge.u-aizu.ac.jp/courses/library/3/DSL/all/DSL_2_I"

#include <bits/stdc++.h>

using namespace std;

#include "../../DataStructure/lazy_segment_tree.cpp"

using ll = long long;
const int Inf = 1e9;

int main() {
  int n, q;
  cin >> n >> q;
  using pi = pair<ll, int>;
  LazySegmentTree<pi, int> seg = LazySegmentTree<pi, int>(n, pi(0LL, 0), Inf, [](pi a, pi b) { return pi(a.first + b.first, a.second + b.second); }, [](pi a, int b) { return pi(a.second * b, a.second); }, [](int a, int b) { return b; }, [](int a, int b) { return a; });
  for (int i = 0; i < n; ++i) seg.set(i, pi(0LL, 1));
  seg.build();
  for (int i = 0; i < q; ++i) {
	int c, s, t;
	cin >> c >> s >> t;
	if (c) {
	  cout << seg.query(s, t).first << endl;
	} else {
	  int x;
	  cin >> x;
	  seg.update(s, t, x);
	}
  }

  return 0;
}

#line 1 "test/aoj/DSL_2_I.test.cpp"
#define PROBLEM "https://onlinejudge.u-aizu.ac.jp/courses/library/3/DSL/all/DSL_2_I"

#include <bits/stdc++.h>

using namespace std;

#line 1 "DataStructure/lazy_segment_tree.cpp"



#line 5 "DataStructure/lazy_segment_tree.cpp"

using namespace std;

template <typename T, typename U = T>
class LazySegmentTree {
 public:
  int n;
  T e;
  U oe;
  vector<T> node;
  vector<U> lazy;

  using F = function<T(T, T)>;
  using G = function<T(T, U)>;
  using H = function<U(U, U)>;
  using P = function<U(U, int)>;

  const F f;
  const G g;
  const H h;
  const P p;

  LazySegmentTree() {}
  LazySegmentTree(int n_, const T e_, const U oe_, const F f_, const G g_,
                  const H h_, const P p_)
      : e(e_), oe(oe_), f(f_), g(g_), h(h_), p(p_) {
    n = 1;
    while (n < n_) n <<= 1;
    node.assign(2 * n, e);
    lazy.assign(2 * n, oe);
  }

  void build() {
    for (int i = n - 1; i > 0; --i) {
      node[i] = f(node[i * 2 + 0], node[i * 2 + 1]);
    }
  }

  void set(int idx, T v) { node[idx + n] = v; }

  void eval(int k, int len) {
    if (lazy[k] == oe) return;
    if (k < n) {
      lazy[k * 2 + 0] = h(lazy[k * 2 + 0], lazy[k]);
      lazy[k * 2 + 1] = h(lazy[k * 2 + 1], lazy[k]);
    }
    node[k] = g(node[k], p(lazy[k], len));
    lazy[k] = oe;
  }

  void update(int a, int b, const U x) { update(a, b + 1, x, 1, 0, n); }

  void update(int a, int b, U x, int k, int l, int r) {
    eval(k, r - l);
    if (a <= l && b >= r) {
      lazy[k] = h(lazy[k], x);
      eval(k, r - l);
    } else if (a < r && b > l) {
      update(a, b, x, k * 2 + 0, l, (l + r) >> 1);
      update(a, b, x, k * 2 + 1, (l + r) >> 1, r);
      node[k] = f(node[k * 2 + 0], node[k * 2 + 1]);
    }
  }

  T query(int a, int b) { return query(a, b + 1, 1, 0, n); }

  T query(int a, int b, int k, int l, int r) {
    eval(k, r - l);
    if (a >= r || b <= l) return e;
    if (a <= l && b >= r) return node[k];
    T vl = query(a, b, k * 2 + 0, l, (l + r) >> 1);
    T vr = query(a, b, k * 2 + 1, (l + r) >> 1, r);
    return f(vl, vr);
  }

  T operator[](int idx) { return query(idx, idx + 1); }
};


#line 8 "test/aoj/DSL_2_I.test.cpp"

using ll = long long;
const int Inf = 1e9;

int main() {
  int n, q;
  cin >> n >> q;
  using pi = pair<ll, int>;
  LazySegmentTree<pi, int> seg = LazySegmentTree<pi, int>(n, pi(0LL, 0), Inf, [](pi a, pi b) { return pi(a.first + b.first, a.second + b.second); }, [](pi a, int b) { return pi(a.second * b, a.second); }, [](int a, int b) { return b; }, [](int a, int b) { return a; });
  for (int i = 0; i < n; ++i) seg.set(i, pi(0LL, 1));
  seg.build();
  for (int i = 0; i < q; ++i) {
	int c, s, t;
	cin >> c >> s >> t;
	if (c) {
	  cout << seg.query(s, t).first << endl;
	} else {
	  int x;
	  cin >> x;
	  seg.update(s, t, x);
	}
  }

  return 0;
}