Skip to the content.

:heavy_check_mark: Count Square Free
(number-theory/count-square-free.hpp)

$N$ 以下の無平方数の個数を $O(N^{2/5}\log\log N)$ 時間で求める.

アルゴリズム

参考:Counting square free numbers - Blog of smsxgz

$N$ 以下の無平方数の個数を $S(N)$ とする.

$~O(\sqrt{N})$

素因数についての包除で次を得る. \(S(N)=\sum_{d}\mu(d)\left\lfloor\frac{N}{d^2}\right\rfloor\) $n$ 以下に対するMobius関数の列挙は篩で $~O(n)$ なのでこれは $~O(\sqrt{N})$ で計算できる.

高速化

$D$ を十分大きくとると,$\left\lfloor\frac{N}{d^2}\right\rfloor$ が $d\gt D$ の範囲であまり変化しなくなる.

$S_1(N)=\sum_{1\leq d\leq D}\mu(d)\left\lfloor\frac{N}{d^2}\right\rfloor,S_2(N)=\sum_{d\gt D}\mu(d)\left\lfloor\frac{N}{d^2}\right\rfloor$ とする.

$S_1$ は直接計算もできるが,$S_2$ でも使えるので $\mu$ 関数の値を列挙してそこから計算するようにする. $\mu(1),\dots,\mu(D)$ の列挙は篩を考えると空間 $O(D)$,時間 $O(D\log\log D)$ で計算できる.

$S_2$ を考察. \(\begin{align*} S_2(N) &=\sum_{d\gt D}\mu(d)\sum_i1_{i=\left\lfloor\frac{N}{d^2}\right\rfloor}i\\ &=\sum_i i\sum_{d\gt D}\mu(d)1_{i\leq\frac{N}{d^2}\lt i+1}\\ &=\sum_i i\sum_{d\gt D,\left\lfloor\sqrt{\frac{N}{i+1}}\right\rfloor\lt d\leq\left\lfloor\sqrt{\frac{N}{i}}\right\rfloor}\mu(d) \end{align*}\)

$x_i=\sqrt{\frac{N}{i}}$ とおく.正整数 $I$ をとり $D=\lfloor x_I\rfloor$ とすれば,Mertens関数 $M(x)=\sum_{i=1}^{\lfloor x\rfloor}\mu(i)$ を用いて次のように表せる. \(S_2(N) =\sum_{i=1}^{I-1}i(M(x_i)-M(x_{i+1})) =\sum_{i=1}^{I-1}M(x_i)-(I-1)M(x_I)\)

ここでメビウス反転公式から $\sum_{n=1}^{x}M\left(\frac{x}{n}\right)=1$ である. 変形すれば $M(x)=1-\sum_{n=2}^{x}M\left(\frac{x}{n}\right)$.

特に $M(x/n),n\geq 2$ の値がわかっているとき $M(x)$ が $O(\sqrt{x})$ で計算できる.

$M(x_I),M(x_{I-1}),\dots,M(x_1)$ をこの順に計算していくことを考える.

$M(x_k)$ を計算するには $M(x_k/i),2\leq i\leq x_k$ の値が必要.

$M(x_k/i)$ は整数部分 $\lfloor x_k/i\rfloor$ で決まるので $M(x_k)$ は $O(\sqrt{x_k})$ で計算できる.

計算量を解析する.

$S_1$ の計算量は $O(D\log\log D)=O(\sqrt{N/I}\log\log(N/I))$.

$S_2$ の計算量は,次が成り立つから $O(N^{1/4}I^{3/4})$. \(\sum_{k=1}^{I}\sqrt{x_k} =\sum_{k=1}^{I}\frac{N^{1/4}}{k^{1/4}} =O(N^{1/4}I^{3/4})\)

ここで $I=N^\alpha$ とおけば上の計算量はそれぞれ $O(N^{(1-\alpha)/2}\log\log N)$ および $O(N^{1/4+3/4\alpha})$ になる. $\alpha=1/5$ とすれば全体の計算量は $O(N^{2/5}\log\log N)$ となった(このとき $D=N^{2/5}$).

Depends on

Verified with

Code

#pragma once

#include "number-theory/mobius-function.hpp"
#include "math/util.hpp"

long long CountSquareFree(long long n) {
  using ll = long long;
  if (n <= 0) return 0;
  ll ret = 0;
  if (n < 100) {
    auto mu = MobiusFunction::table((int)n);
    for (int i = 1; i <= n; i++) ret += n / i / i * mu[i];
  } else {
    ll l = Math::floor_root(n, 5);
    ll d = Math::floor_root(n / l, 2);
    auto mu = MobiusFunction::table(d);
    for (int i = 1; i <= d; i++) ret += n / i / i * mu[i];
    for (int i = 1; i <= d; i++) mu[i] += mu[i - 1];  // mu 自体はもういらないので使い回し
    vector<ll> a(l + 1);
    for (int k = l; k > 0; k--) {
      ll x = Math::floor_root(n / k, 2);
      ll sq = Math::floor_root(x, 2);
      ll m = x / (sq + 1);
      ll v = 1;
      for (ll i = 1; i <= m; i++) v -= (x / i - x / (i + 1)) * mu[i];
      for (ll i = 2; i <= sq; i++) v -= x / i <= d ? mu[x / i] : a[k * i * i];
      a[k] = v;
      if (k < l) ret += a[k];
    }
    ret -= (l - 1) * a[l];
  }
  return ret;
}

/**
 * @brief Count Square Free
 * @docs docs/number-theory/count-square-free.md
 */
#line 2 "number-theory/count-square-free.hpp"

#line 2 "number-theory/mobius-function.hpp"

#line 2 "math/util.hpp"

namespace Math {
template <class T>
T safe_mod(T a, T b) {
  assert(b != 0);
  if (b < 0) a = -a, b = -b;
  a %= b;
  return a >= 0 ? a : a + b;
}
template <class T>
T floor(T a, T b) {
  assert(b != 0);
  if (b < 0) a = -a, b = -b;
  return a >= 0 ? a / b : (a + 1) / b - 1;
}
template <class T>
T ceil(T a, T b) {
  assert(b != 0);
  if (b < 0) a = -a, b = -b;
  return a > 0 ? (a - 1) / b + 1 : a / b;
}
long long isqrt(long long n) {
  if (n <= 0) return 0;
  long long x = sqrt(n);
  while ((__int128)(x + 1) * (x + 1) <= n) x++;
  while ((__int128)x * x > n) x--;
  return x;
}
long long floor_root(long long n, int k) {
  assert(n >= 0);
  if (n == 0) return 0;
  assert(k >= 1);
  if (k == 1) return n;
  if (k > 64) return 1;
  long long x = round(pow((long double)n, 1.0L / k));
  auto check = [&](long long a) {
    if (a <= 0) return true;
    __int128_t p = 1;
    for (int i = 0; i < k; ++i)
      if ((p *= a) > n) return false;
    return true;
  };
  while (check(x + 1)) x++;
  while (!check(x)) x--;
  return x;
}
unsigned long long floor_root_unsigned(unsigned long long n, int k) {
  assert(k >= 1);
  if (n <= 1 || k == 1) return n;
  if (k >= 64) return 1;
  int bits = (64 + k - 1) / k;
  unsigned long long ok = 1, ng = min(n, 1ULL << bits);
  auto check = [&](unsigned long long a) {
    __uint128_t p = 1;
    for (int i = 0; i < k; i++) {
      p *= a;
      if (p > n) return false;
    }
    return true;
  };
  while (ok + 1 < ng) {
    unsigned long long mid = ok + (ng - ok) / 2;
    (check(mid) ? ok : ng) = mid;
  }
  return ok;
}
// return g=gcd(a,b)
// a*x+b*y=g
// - b!=0 -> 0<=x<|b|/g
// - b=0  -> ax=g
template <class T>
T ext_gcd(T a, T b, T& x, T& y) {
  T a0 = a, b0 = b;
  bool sgn_a = a < 0, sgn_b = b < 0;
  if (sgn_a) a = -a;
  if (sgn_b) b = -b;
  if (b == 0) {
    x = sgn_a ? -1 : 1;
    y = 0;
    return a;
  }
  T x00 = 1, x01 = 0, x10 = 0, x11 = 1;
  while (b != 0) {
    T q = a / b, r = a - b * q;
    x00 -= q * x01;
    x10 -= q * x11;
    swap(x00, x01);
    swap(x10, x11);
    a = b, b = r;
  }
  x = x00, y = x10;
  if (sgn_a) x = -x;
  if (sgn_b) y = -y;
  if (b0 != 0) {
    a0 /= a, b0 /= a;
    if (b0 < 0) a0 = -a0, b0 = -b0;
    T q = x >= 0 ? x / b0 : (x + 1) / b0 - 1;
    x -= b0 * q;
    y += a0 * q;
  }
  return a;
}
constexpr long long inv_mod(long long x, long long m) {
  x %= m;
  if (x < 0) x += m;
  long long a = m, b = x;
  long long y0 = 0, y1 = 1;
  while (b > 0) {
    long long q = a / b;
    swap(a -= q * b, b);
    swap(y0 -= q * y1, y1);
  }
  if (y0 < 0) y0 += m / a;
  return y0;
}
long long pow_mod(long long x, long long n, long long m) {
  if (m == 1) return 0;
  x = (x % m + m) % m;
  long long y = 1;
  while (n) {
    if (n & 1) y = y * x % m;
    x = x * x % m;
    n >>= 1;
  }
  return y;
}
constexpr long long pow_mod_constexpr(long long x, long long n, int m) {
  if (m == 1) return 0;
  unsigned int _m = (unsigned int)(m);
  unsigned long long r = 1;
  unsigned long long y = x % m;
  if (y >= m) y += m;
  while (n) {
    if (n & 1) r = (r * y) % _m;
    y = (y * y) % _m;
    n >>= 1;
  }
  return r;
}
constexpr bool is_prime_constexpr(int n) {
  if (n <= 1) return false;
  if (n == 2 || n == 7 || n == 61) return true;
  if (n % 2 == 0) return false;
  long long d = n - 1;
  while (d % 2 == 0) d /= 2;
  constexpr long long bases[3] = {2, 7, 61};
  for (long long a : bases) {
    long long t = d;
    long long y = pow_mod_constexpr(a, t, n);
    while (t != n - 1 && y != 1 && y != n - 1) {
      y = y * y % n;
      t <<= 1;
    }
    if (y != n - 1 && t % 2 == 0) {
      return false;
    }
  }
  return true;
}
template <int n>
constexpr bool is_prime = is_prime_constexpr(n);
};  // namespace Math
#line 2 "number-theory/prime-sieve.hpp"

namespace PrimeSieve {
using ll = long long;

vector<int> lpf(int n) {
  assert(n >= 0);
  vector<int> ret(n + 1);
  for (size_t i = 0; i < ret.size(); i++) ret[i] = (int)i;
  for (int p = 2; (ll)p * p <= n; p++) {
    if (ret[p] != p) continue;
    for (ll x = (ll)p * p;; x += p) {
      if (ret[x] == x) ret[x] = p;
      if (n - x < p) break;
    }
  }
  return ret;
}

vector<int> table(int n) {
  assert(n >= 0);
  vector<bool> composite(n + 1, false);
  for (int p = 2; (ll)p * p <= n; p += (p & 1) + 1) {
    if (composite[p]) continue;
    for (ll x = (ll)p * p;; x += p) {
      composite[x] = true;
      if (n - x < p) break;
    }
  }
  vector<int> ps;
  for (int p = 2; p <= n;) {
    if (!composite[p]) ps.push_back(p);
    int step = (p & 1) + 1;
    if (n - p < step) break;
    p += step;
  }
  return ps;
}

vector<vector<pair<ll, int>>> factorize(int n) {
  assert(n >= 0);
  vector<vector<pair<ll, int>>> factors(n + 1);
  auto lp = lpf(n);
  for (int x = 2; x <= n;) {
    int y = x;
    while (y > 1) {
      int p = lp[y], e = 0;
      while (y % p == 0) y /= p, e++;
      factors[x].emplace_back(p, e);
    }
    if (x == n) break;
    x++;
  }
  return factors;
}
};  // namespace PrimeSieve

/**
 * @brief 素数篩
 * @docs docs/number-theory/prime-sieve.md
 */
#line 5 "number-theory/mobius-function.hpp"

namespace MobiusFunction {
using ll = long long;

vector<int> table(int n) {
  vector<int> mu(n + 1, 1);
  mu[0] = 0;
  auto lpf = PrimeSieve::lpf(n);
  for (int x = 2; x <= n; x++) {
    int p = lpf[x];
    if (x / p % p == 0)
      mu[x] = 0;
    else
      mu[x] = -mu[x / p];
  }
  return mu;
}

ll sum(ll n) {
  if (n <= 0) return 0;

  ll k = ceil(pow((long double)n, 2.0L / 3.0L));
  __int128 n2 = (__int128)n * n;
  auto enough = [&](ll x) { return (__int128)x * x * x >= n2; };
  while (k > 1 && enough(k - 1)) k--;
  while (!enough(k)) k++;
  assert(k <= numeric_limits<int>::max());

  int lim = (int)k;
  vector<ll> small(lim + 1);
  auto mu = table(lim);
  for (int i = 1; i <= lim; i++) small[i] = small[i - 1] + mu[i];

  ll len = n / k + (n % k != 0);
  vector<ll> large(len + 1);
  for (ll i = len; i >= 1; i--) {
    ll x = n / i;
    ll m = Math::isqrt(x);
    ll v = 1;
    for (ll j = 2; j <= m; j++) v -= i * j <= len ? large[i * j] : small[x / j];
    for (ll j = 1; j <= m; j++) v -= (x / j - m) * (small[j] - small[j - 1]);
    large[i] = v;
  }
  return large[1];
}
};  // namespace MobiusFunction

/**
 * @brief Mobius Function
 * @docs docs/number-theory/mobius-function.md
 */
#line 5 "number-theory/count-square-free.hpp"

long long CountSquareFree(long long n) {
  using ll = long long;
  if (n <= 0) return 0;
  ll ret = 0;
  if (n < 100) {
    auto mu = MobiusFunction::table((int)n);
    for (int i = 1; i <= n; i++) ret += n / i / i * mu[i];
  } else {
    ll l = Math::floor_root(n, 5);
    ll d = Math::floor_root(n / l, 2);
    auto mu = MobiusFunction::table(d);
    for (int i = 1; i <= d; i++) ret += n / i / i * mu[i];
    for (int i = 1; i <= d; i++) mu[i] += mu[i - 1];  // mu 自体はもういらないので使い回し
    vector<ll> a(l + 1);
    for (int k = l; k > 0; k--) {
      ll x = Math::floor_root(n / k, 2);
      ll sq = Math::floor_root(x, 2);
      ll m = x / (sq + 1);
      ll v = 1;
      for (ll i = 1; i <= m; i++) v -= (x / i - x / (i + 1)) * mu[i];
      for (ll i = 2; i <= sq; i++) v -= x / i <= d ? mu[x / i] : a[k * i * i];
      a[k] = v;
      if (k < l) ret += a[k];
    }
    ret -= (l - 1) * a[l];
  }
  return ret;
}

/**
 * @brief Count Square Free
 * @docs docs/number-theory/count-square-free.md
 */
Back to top page