This documentation is automatically generated by online-judge-tools/verification-helper
View the Project on GitHub knshnb/competitive_library
#include "src/DataStructure/LazySegmentTree.hpp"
区間更新区間取得セグメント木。
参考
f0(T0, T0) -> T0
f1(T1, T1) -> T1
g(T0, T1) -> T0
p(T1 x, int len) -> T1
query(int l, int r) -> T
update(int l, int t, T1 x)
Int
auto seg = make_lazy_segment_tree<Int, Int>(std::plus<Int>(), 0, std::plus<Int>(), 0, std::plus<Int>(), [](Int x, int len) { return x * len; });
constexpr Int INF = 1e18; auto seg = make_lazy_segment_tree<Int, Int>( [](Int x, Int y) { return std::max(x, y); }, -INF, [](Int x, Int y) { return x + y; }, 0, [](Int x, Int y) { return x == -INF ? x : x + y; }, [](Int x, int len) { return x; });
constexpr Int INF = 1e18; auto seg = make_lazy_segment_tree<Int, Int>( [](Int x, Int y) { return std::min(x, y); }, INF, [](Int x, Int y) { return x + y; }, 0, [](Int x, Int y) { return x == INF ? x : x + y; }, [](Int x, int len) { return x; });
mint
auto seg = make_lazy_segment_tree<mint, pmm>( std::plus<mint>(), 0, [](pmm x, pmm y) -> pmm { return {x.first * y.first, x.second * y.first + y.second}; }, {1, 0}, [](mint x, pmm y) { return y.first * x + y.second; }, [](pmm x, int len) -> pmm { return {x.first, x.second * len}; });
/// @docs src/DataStructure/LazySegmentTree.md template <class T0, class T1, class F0, class F1, class G, class P> class LazySegmentTree { // k番目のノードにのlazyを伝搬 void eval(int k, int len) { // 定数倍高速化 if (lazy[k] == u1) return; // len個分のlazy[k]を評価 node[k] = g(node[k], p(lazy[k], len)); if (k < N - 1) { // 最下段でなければ下のlazyに伝搬 lazy[2 * k + 1] = f1(lazy[2 * k + 1], lazy[k]); lazy[2 * k + 2] = f1(lazy[2 * k + 2], lazy[k]); } lazy[k] = u1; } // k番目のノード[l, r)について、[a, b)の範囲内にxを作用 void update(int a, int b, T1 x, int k, int l, int r) { eval(k, r - l); if (b <= l || r <= a) return; if (a <= l && r <= b) { lazy[k] = f1(lazy[k], x); eval(k, r - l); } else { update(a, b, x, 2 * k + 1, l, (l + r) / 2); update(a, b, x, 2 * k + 2, (l + r) / 2, r); node[k] = f0(node[2 * k + 1], node[2 * k + 2]); } } // k番目のノード[l, r)について、[a, b)のクエリを求める T0 query(int a, int b, int k, int l, int r) { if (r <= a || b <= l) return u0; eval(k, r - l); if (a <= l && r <= b) return node[k]; T0 vl = query(a, b, 2 * k + 1, l, (l + r) / 2); T0 vr = query(a, b, 2 * k + 2, (l + r) / 2, r); return f0(vl, vr); } public: int sz; // 元の配列のサイズ int N; std::vector<T0> node; std::vector<T1> lazy; // T0上の演算、単位元 const F0 f0; const T0 u0; // T1上の演算、単位元 const F1 f1; const T1 u1; // T0に対するT1の作用 const G g; // 多数のt1(T1)に対するf1の合成 const P p; LazySegmentTree(F0 f0_, T0 u0_, F1 f1_, T1 u1_, G g_, P p_) : f0(f0_), u0(u0_), f1(f1_), u1(u1_), g(g_), p(p_) {} void set_by_vector(const std::vector<T0>& a) { sz = a.size(); for (N = 1; N < sz; N *= 2) ; node.resize(2 * N - 1, u0); lazy.resize(2 * N - 1, u1); for (int i = 0; i < sz; i++) node[N - 1 + i] = a[i]; for (int i = N - 2; i >= 0; i--) node[i] = f0(node[2 * i + 1], node[2 * i + 2]); } void set_by_unit(int n) { sz = n; for (N = 1; N < sz; N *= 2) ; node.resize(2 * N - 1, u0); lazy.resize(2 * N - 1, u1); } // [a, b)にxを作用 void update(int a, int b, T1 x) { assert(0 <= a && a < b && b <= sz); update(a, b, x, 0, 0, N); } void update(int a, T1 x) { update(a, a + 1, x); } // [a, b) T0 query(int a, int b) { return query(a, b, 0, 0, N); } T0 query(int a) { return query(a, a + 1); } friend std::string to_string(LazySegmentTree<T0, T1, F0, F1, G, P>& seg) { for (int i = 0; i < seg.sz; i++) seg.query(i); return to_string(std::vector<T0>(seg.node.begin() + (seg.N - 1), seg.node.begin() + (seg.N - 1 + seg.sz))); } }; template <class T0, class T1, class F0, class F1, class G, class P> auto make_lazy_segment_tree(F0 f0, T0 u0, F1 f1, T1 u1, G g, P p) { return LazySegmentTree<T0, T1, F0, F1, G, P>(f0, u0, f1, u1, g, p); }
#line 1 "src/DataStructure/LazySegmentTree.hpp" /// @docs src/DataStructure/LazySegmentTree.md template <class T0, class T1, class F0, class F1, class G, class P> class LazySegmentTree { // k番目のノードにのlazyを伝搬 void eval(int k, int len) { // 定数倍高速化 if (lazy[k] == u1) return; // len個分のlazy[k]を評価 node[k] = g(node[k], p(lazy[k], len)); if (k < N - 1) { // 最下段でなければ下のlazyに伝搬 lazy[2 * k + 1] = f1(lazy[2 * k + 1], lazy[k]); lazy[2 * k + 2] = f1(lazy[2 * k + 2], lazy[k]); } lazy[k] = u1; } // k番目のノード[l, r)について、[a, b)の範囲内にxを作用 void update(int a, int b, T1 x, int k, int l, int r) { eval(k, r - l); if (b <= l || r <= a) return; if (a <= l && r <= b) { lazy[k] = f1(lazy[k], x); eval(k, r - l); } else { update(a, b, x, 2 * k + 1, l, (l + r) / 2); update(a, b, x, 2 * k + 2, (l + r) / 2, r); node[k] = f0(node[2 * k + 1], node[2 * k + 2]); } } // k番目のノード[l, r)について、[a, b)のクエリを求める T0 query(int a, int b, int k, int l, int r) { if (r <= a || b <= l) return u0; eval(k, r - l); if (a <= l && r <= b) return node[k]; T0 vl = query(a, b, 2 * k + 1, l, (l + r) / 2); T0 vr = query(a, b, 2 * k + 2, (l + r) / 2, r); return f0(vl, vr); } public: int sz; // 元の配列のサイズ int N; std::vector<T0> node; std::vector<T1> lazy; // T0上の演算、単位元 const F0 f0; const T0 u0; // T1上の演算、単位元 const F1 f1; const T1 u1; // T0に対するT1の作用 const G g; // 多数のt1(T1)に対するf1の合成 const P p; LazySegmentTree(F0 f0_, T0 u0_, F1 f1_, T1 u1_, G g_, P p_) : f0(f0_), u0(u0_), f1(f1_), u1(u1_), g(g_), p(p_) {} void set_by_vector(const std::vector<T0>& a) { sz = a.size(); for (N = 1; N < sz; N *= 2) ; node.resize(2 * N - 1, u0); lazy.resize(2 * N - 1, u1); for (int i = 0; i < sz; i++) node[N - 1 + i] = a[i]; for (int i = N - 2; i >= 0; i--) node[i] = f0(node[2 * i + 1], node[2 * i + 2]); } void set_by_unit(int n) { sz = n; for (N = 1; N < sz; N *= 2) ; node.resize(2 * N - 1, u0); lazy.resize(2 * N - 1, u1); } // [a, b)にxを作用 void update(int a, int b, T1 x) { assert(0 <= a && a < b && b <= sz); update(a, b, x, 0, 0, N); } void update(int a, T1 x) { update(a, a + 1, x); } // [a, b) T0 query(int a, int b) { return query(a, b, 0, 0, N); } T0 query(int a) { return query(a, a + 1); } friend std::string to_string(LazySegmentTree<T0, T1, F0, F1, G, P>& seg) { for (int i = 0; i < seg.sz; i++) seg.query(i); return to_string(std::vector<T0>(seg.node.begin() + (seg.N - 1), seg.node.begin() + (seg.N - 1 + seg.sz))); } }; template <class T0, class T1, class F0, class F1, class G, class P> auto make_lazy_segment_tree(F0 f0, T0 u0, F1 f1, T1 u1, G g, P p) { return LazySegmentTree<T0, T1, F0, F1, G, P>(f0, u0, f1, u1, g, p); }