This documentation is automatically generated by online-judge-tools/verification-helper
View the Project on GitHub noshi91/Library
#define PROBLEM "https://judge.yosupo.jp/problem/range_affine_range_sum" #include "data_structure/lazy_segment_tree.cpp" #include "other/modint.cpp" #include "other/sum_affine_action.cpp" #include <iostream> int main() { #include "other/fast_ios.cpp" using mint = modint<998244353>; int n, q; std::cin >> n >> q; lazy_segment_tree<sum_affine_action<mint>> lst(n); for (int i = 0; i != n; i += 1) { int a; std::cin >> a; lst.update_point(i, {a, 1}); } for (int i = 0; i != q; i += 1) { int t; std::cin >> t; switch (t) { case 0: { int l, r, b, c; std::cin >> l >> r >> b >> c; lst.update_range(l, r, {b, c}); } break; case 1: { int l, r; std::cin >> l >> r; std::cout << lst.fold(l, r).first.value() << "\n"; } break; } } }
#line 1 "test/lazy_segment_tree.test.cpp" #define PROBLEM "https://judge.yosupo.jp/problem/range_affine_range_sum" #line 2 "other/bit_width.cpp" #line 2 "other/countl_zero.cpp" #line 2 "other/countr_zero.cpp" #line 2 "other/int_alias.cpp" #include <cstddef> #include <cstdint> using i32 = std::int32_t; using i64 = std::int64_t; using u32 = std::uint32_t; using u64 = std::uint64_t; using isize = std::ptrdiff_t; using usize = std::size_t; #line 4 "other/countr_zero.cpp" #include <array> usize countr_zero(u64 x) { if (x == 0) return 64; #ifdef __GNUC__ return __builtin_ctzll(x); #else constexpr std::array<usize, 64> table = { 0, 1, 2, 7, 3, 13, 8, 27, 4, 33, 14, 36, 9, 49, 28, 19, 5, 25, 34, 17, 15, 53, 37, 55, 10, 46, 50, 39, 29, 42, 20, 57, 63, 6, 12, 26, 32, 35, 48, 18, 24, 16, 52, 54, 45, 38, 41, 56, 62, 11, 31, 47, 23, 51, 44, 40, 61, 30, 22, 43, 60, 21, 59, 58}; return table[(x & ~x + 1) * 0x218A7A392DD9ABF >> 58 & 0x3F]; #endif } #line 5 "other/countl_zero.cpp" usize countl_zero(u64 x) { #ifdef __GNUC__ return x == 0 ? 64 : __builtin_clzll(x); #else x |= x >> 1; x |= x >> 2; x |= x >> 4; x |= x >> 8; x |= x >> 16; x |= x >> 32; return 64 - countr_zero(~x); #endif } #line 5 "other/bit_width.cpp" usize bit_width(const u64 x) { return 64 - countl_zero(x); } #line 4 "data_structure/lazy_segment_tree.cpp" #include <cassert> #line 7 "data_structure/lazy_segment_tree.cpp" #include <vector> template <class A> class lazy_segment_tree { using V = typename A::value_structure; using T = typename V::value_type; using O = typename A::operator_structure; using E = typename O::value_type; class node_type { public: T value; E lazy; node_type(const T value, const E lazy) : value(value), lazy(lazy) {} }; static T get(const node_type &x) { return A::operation(x.value, x.lazy); } static void add(E &x, const E y) { x = O::operation(x, y); } std::vector<node_type> tree; void propagate(const usize index) { add(tree[index * 2].lazy, tree[index].lazy); add(tree[index * 2 + 1].lazy, tree[index].lazy); tree[index].lazy = O::identity; } void propagate_bound(const usize index) { if (index == 0) return; const usize crz = countr_zero(index); for (usize h = bit_width(index) - 1; h != crz; h -= 1) propagate(index >> h); } void recalc(const usize index) { tree[index].value = V::operation(get(tree[index * 2]), get(tree[index * 2 + 1])); } void recalc_bound(usize index) { if (index == 0) return; index >>= countr_zero(index); while (index != 1) { index /= 2; recalc(index); } } public: lazy_segment_tree() = default; explicit lazy_segment_tree(const usize n) : tree(n * 2, node_type(V::identity, O::identity)) {} usize size() const { return tree.size() / 2; } T fold(usize first, usize last) { assert(first <= last); assert(last <= size()); first += size(); last += size(); propagate_bound(first); recalc_bound(first); propagate_bound(last); recalc_bound(last); T fold_l = V::identity; T fold_r = V::identity; while (first != last) { if (first % 2 != 0) { fold_l = V::operation(fold_l, get(tree[first])); first += 1; } first /= 2; if (last % 2 != 0) { last -= 1; fold_r = V::operation(get(tree[last]), fold_r); } last /= 2; } return V::operation(fold_l, fold_r); } void update_range(usize first, usize last, const E x) { assert(first <= last); assert(last <= size()); first += size(); last += size(); propagate_bound(first); propagate_bound(last); const usize first_c = first; const usize last_c = last; while (first != last) { if (first % 2 != 0) { add(tree[first].lazy, x); first += 1; } first /= 2; if (last % 2 != 0) { last -= 1; add(tree[last].lazy, x); } last /= 2; } recalc_bound(first_c); recalc_bound(last_c); } void update_point(usize index, const T x) { assert(index < size()); index += size(); for (usize h = bit_width(index) - 1; h != 0; h -= 1) propagate(index >> h); tree[index] = node_type(x, O::identity); while (index != 1) { index /= 2; recalc(index); } } }; /** * @brief Lazy Segment Tree * @see https://scrapbox.io/data-structures/Lazy_Segment_Tree */ #line 2 "other/modint.cpp" template <std::uint_fast64_t Modulus> class modint { using u64 = std::uint_fast64_t; public: using value_type = u64; static constexpr u64 mod = Modulus; private: static_assert(mod < static_cast<u64>(1) << 32, "Modulus must be less than 2**32"); u64 v; constexpr modint &negate() noexcept { if (v != 0) v = mod - v; return *this; } public: constexpr modint(const u64 x = 0) noexcept : v(x % mod) {} constexpr u64 &value() noexcept { return v; } constexpr const u64 &value() const noexcept { return v; } constexpr modint operator+() const noexcept { return modint(*this); } constexpr modint operator-() const noexcept { return modint(*this).negate(); } constexpr modint operator+(const modint rhs) const noexcept { return modint(*this) += rhs; } constexpr modint operator-(const modint rhs) const noexcept { return modint(*this) -= rhs; } constexpr modint operator*(const modint rhs) const noexcept { return modint(*this) *= rhs; } constexpr modint operator/(const modint rhs) const noexcept { return modint(*this) /= rhs; } constexpr modint &operator+=(const modint rhs) noexcept { v += rhs.v; if (v >= mod) v -= mod; return *this; } constexpr modint &operator-=(const modint rhs) noexcept { if (v < rhs.v) v += mod; v -= rhs.v; return *this; } constexpr modint &operator*=(const modint rhs) noexcept { v = v * rhs.v % mod; return *this; } constexpr modint &operator/=(modint rhs) noexcept { u64 exp = mod - 2; while (exp != 0) { if (exp % 2 != 0) *this *= rhs; rhs *= rhs; exp /= 2; } return *this; } }; template <std::uint_fast64_t Modulus> constexpr typename modint<Modulus>::u64 modint<Modulus>::mod; #line 1 "other/affine.cpp" template <class T> class affine { public: T a; T b; constexpr affine(const T &a = 1, const T &b = 0) noexcept : a(a), b(b) {} constexpr T evaluate(const T &x) const noexcept { return x * a + b; } friend constexpr affine operator+(const affine &l, const affine &r) noexcept { return affine(l.a + r.a, l.b + r.b); } constexpr affine composite(const affine &r) const noexcept { return affine(a * r.a, b * r.a + r.b); } }; template <class T> class affine_composite_monoid { public: using value_type = affine<T>; static constexpr value_type operation(const value_type &l, const value_type &r) noexcept { return l.composite(r); } static constexpr value_type identity{}; }; #line 1 "other/cartesian_product_monoid.cpp" #include <utility> template <class M, class N> class cartesian_product_monoid { using T = std::pair<typename M::value_type, typename N::value_type>; public: using value_type = T; static constexpr T operation(const T &l, const T &r) noexcept { return T(M::operation(l.first, r.first), N::operation(l.second, r.second)); } static constexpr T identity{M::identity, N::identity}; }; #line 1 "other/plus_monoid.cpp" template <class T> class plus_monoid { public: using value_type = T; static T operation(const T l, const T r) { return l + r; } static constexpr T identity = 0; }; #line 4 "other/sum_affine_action.cpp" template <class T> class sum_affine_action { public: using value_structure = cartesian_product_monoid<plus_monoid<T>, plus_monoid<T>>; using operator_structure = affine_composite_monoid<T>; private: using U = typename value_structure::value_type; using E = typename operator_structure::value_type; public: static constexpr U operation(const U &l, const E &r) { return U(l.first * r.a + l.second * r.b, l.second); } }; #line 6 "test/lazy_segment_tree.test.cpp" #include <iostream> int main() { #line 1 "other/fast_ios.cpp" std::ios::sync_with_stdio(false); std::cin.tie(nullptr); #line 11 "test/lazy_segment_tree.test.cpp" using mint = modint<998244353>; int n, q; std::cin >> n >> q; lazy_segment_tree<sum_affine_action<mint>> lst(n); for (int i = 0; i != n; i += 1) { int a; std::cin >> a; lst.update_point(i, {a, 1}); } for (int i = 0; i != q; i += 1) { int t; std::cin >> t; switch (t) { case 0: { int l, r, b, c; std::cin >> l >> r >> b >> c; lst.update_range(l, r, {b, c}); } break; case 1: { int l, r; std::cin >> l >> r; std::cout << lst.fold(l, r).first.value() << "\n"; } break; } } }