alg/acted_monoid/cntsum_add.hpp

View this file on GitHub
Last update: 2024-01-23 05:58:02+09:00
Include: #include "alg/acted_monoid/cntsum_add.hpp"

Depends on

Verified with

test/yukicoder/235.test.cpp

Code

#include "alg/monoid/add_pair.hpp"
#include "alg/monoid/add.hpp"

template <typename E>
struct ActedMonoid_CntSum_Add {
  using Monoid_X = Monoid_Add_Pair<E>;
  using Monoid_A = Monoid_Add<E>;
  using X = typename Monoid_X::value_type;
  using A = typename Monoid_A::value_type;
  static constexpr X act(const X &x, const A &a, const ll &size) {
    return {x.fi, x.se + x.fi * a};
  }
};

#line 2 "alg/monoid/add_pair.hpp"

template <typename E>
struct Monoid_Add_Pair {
  using value_type = pair<E, E>;
  using X = value_type;
  static constexpr X op(const X &x, const X &y) {
    return {x.fi + y.fi, x.se + y.se};
  }
  static constexpr X inverse(const X &x) { return {-x.fi, -x.se}; }
  static constexpr X unit() { return {0, 0}; }
  static constexpr bool commute = true;
};
#line 2 "alg/monoid/add.hpp"

template <typename E>
struct Monoid_Add {
  using X = E;
  using value_type = X;
  static constexpr X op(const X &x, const X &y) noexcept { return x + y; }
  static constexpr X inverse(const X &x) noexcept { return -x; }
  static constexpr X power(const X &x, ll n) noexcept { return X(n) * x; }
  static constexpr X unit() { return X(0); }
  static constexpr bool commute = true;
};
#line 3 "alg/acted_monoid/cntsum_add.hpp"

template <typename E>
struct ActedMonoid_CntSum_Add {
  using Monoid_X = Monoid_Add_Pair<E>;
  using Monoid_A = Monoid_Add<E>;
  using X = typename Monoid_X::value_type;
  using A = typename Monoid_A::value_type;
  static constexpr X act(const X &x, const A &a, const ll &size) {
    return {x.fi, x.se + x.fi * a};
  }
};