小 Q 与函数求和 1

小 Q 与函数求和 1

所以预先处理次幂，及，即可同时算得，以及，整体复杂度，稍卡常，得写 add sub 函数才能过。

#include <bits/stdc++.h>

using namespace std;

const int N = 5e6 + 10, mod = 998244353;

int prime[N], inv[N], phi[N], mu[N], f[N], g[N], a[N], n, k, cnt;

bool st[N];

inline int add(int x, int y) {
  return x + y < mod ? x + y : x + y - mod;
}

inline void Add(int &x, int y) {
  x + y < mod ? x += y : x += y - mod;
}

inline int sub(int x, int y) {
  return x >= y ? x - y : x - y + mod;
}

inline void Sub(int &x, int y) {
  x >= y ? x -= y : x += mod - y;
}

int quick_pow(int a, int n) {
  int ans = 1;
  while (n) {
    if (n & 1) {
      ans = 1ll * ans * a % mod;
    }
    a = 1ll * a * a % mod;
    n >>= 1;
  }
  return ans;
}

void init() {
  phi[1] = mu[1] = a[1] = inv[1] = 1;
  for (int i = 2; i < N; i++) {
    inv[i] = 1ll * (mod - mod / i) * inv[mod % i] % mod;
    if (!st[i]) {
      prime[++cnt] = i;
      phi[i] = i - 1;
      mu[i] = -1;
      a[i] = quick_pow(i, k);
    }
    for (int j = 1; j <= cnt && 1ll * i * prime[j] < N; j++) {
      st[i * prime[j]] = 1;
      a[i * prime[j]] = 1ll * a[i] * a[prime[j]] % mod;
      if (i % prime[j] == 0) {
        phi[i * prime[j]] = phi[i] * prime[j];
        break;
      }
      phi[i * prime[j]] = phi[i] * (prime[j] - 1);
      mu[i * prime[j]] = -mu[i];
    }
  }
  for (int i = 1; i <= n; i++) {
    for (int j = i; j <= n; j += i) {
      Add(f[i], phi[j]);
      if (mu[j / i] == 1) {
        Add(g[j], 1ll * a[i] * inv[phi[i]] % mod);
      }
      else if (mu[j / i] == -1) {
        Sub(g[j], 1ll * a[i] * inv[phi[i]] % mod);
      }
    }
  }
}

int main() {
  // freopen("in.txt", "r", stdin);
  // freopen("out.txt", "w", stdout);
  scanf("%d %d", &n, &k);
  k++;
  init();
  int ans = 0;
  for (int i = 1; i <= n; i++) {
    Add(ans, 1ll * f[i] * f[i] % mod * g[i] % mod);
  }
  printf("%d\n", ans);
  return 0;
}