Multiple Möbius Transform
(algorithm/multiple_mobius_transform.cpp)

• View this file on GitHub
• Last update: 2020-03-11 23:32:43+09:00

Depends on

• other/int_alias.cpp

Code

#include "other/int_alias.cpp"

#include <vector>

template <class G>
void multiple_mobius_transform(std::vector<typename G::value_type> &a) {
  const usize n = a.size();
  std::vector<bool> sieve(n, true);
  for (usize p = 2; p < n; p += 1) {
    if (!sieve[p])
      continue;
    for (usize k = 1; k * p < n; k += 1) {
      sieve[k * p] = false;
      a[k] = G::operation(a[k], G::inverse(a[k * p]));
    }
  }
}

/**
 * @brief Multiple Möbius Transform
 */

#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 2 "algorithm/multiple_mobius_transform.cpp"

#include <vector>

template <class G>
void multiple_mobius_transform(std::vector<typename G::value_type> &a) {
  const usize n = a.size();
  std::vector<bool> sieve(n, true);
  for (usize p = 2; p < n; p += 1) {
    if (!sieve[p])
      continue;
    for (usize k = 1; k * p < n; k += 1) {
      sieve[k * p] = false;
      a[k] = G::operation(a[k], G::inverse(a[k * p]));
    }
  }
}

/**
 * @brief Multiple Möbius Transform
 */

Back to top page