一.前置知识
莫比乌斯反演
数论分块
二.题意及分析
①题意:给出1~n,求其中子序列能组成等比数列的数目
②初步分析:设公比为,首项为p,末项为q,都为整数。
由于很容易知道q必须要整除a^(k-1)。而这样末项q有
种。
而对于分母上,每一个a,由开头公比定义,还有一个约束条件,就是必须有一个小于a并与a互质的数b,才能与他配对
因此两条件需要同时满足,得出式子:
③数学分析:
a.首先是很经典的杜教筛,这里分块一次链接
b.然后考虑后半部分:
k=1自成等比数列,k=2任意取出两数也必成等比数列,结果易得。
k=3,分母是平方,所以是常规的分块(第二次分块)
k>3,暴力算*phi(a),然后累加求和即可(见代码注释)因为n最大1e17,需要枚举
个,最多也是不到1e6
三、代码
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 40000000 + 10;//小范围会莫名T ,大范围会MLE
const int mod = 998244353;
int pri[maxn], vis[maxn], cnt = 0, phi[maxn], sum[maxn];
void init() {//素数筛
vis[1] = phi[1] = 1;
cnt = 0;
for (int i = 2; i < maxn; i++) {
if (!vis[i]) {
pri[++cnt] = i;
phi[i] = i - 1;
}
for (int j = 1; j <= cnt && i * pri[j] < maxn; j++) {
vis[i * pri[j]] = 1;
if (i % pri[j] == 0) {
phi[i * pri[j]] = phi[i] * pri[j];
break;
}
phi[i * pri[j]] = phi[i] * (pri[j] - 1);
}
}
for (int i = 1; i < maxn; i++) {
sum[i] = (sum[i - 1] + phi[i]) % mod;
}
}
//杜教筛部分
unordered_map<ll, int> mp;
ll get_s1(ll x) {
x %= mod;
ll tmp = x * (x + 1) / 2;
return tmp % mod;
}
ll S_phi(ll x) {//左半部分欧拉函数前缀和
if (x < maxn) return sum[x];
if (mp.count(x)) return mp[x];
ll ans = 0;
for (ll l = 2, r; l <= x; l = r + 1) {
r = x / (x / l);
ans = (ans + (r - l + 1) * S_phi(x / l) % mod) % mod;
}
ans = (get_s1(x) - ans + mod) % mod;
return ans;
}
//欧拉函数前缀和,杜教筛结束
ll get_sqrt(ll a) {
ll x = (ll) sqrt(a);
return x;
}
ll solve(ll n) {
ll ans = (n % mod) * (n % mod + 1) / 2;
//k=1为n,k=2为直接求和
ans %= mod;
//k=3即n/a^2时 ,常规分块
for (ll l = 2, r; l * l <= n; l = r + 1) {
r = get_sqrt(n / (n / l / l));
ans = (ans + n / l / l % mod * (S_phi(r) - S_phi(l - 1) + mod) % mod) % mod;
}
//k>3,n/a^(k-1 ),暴力求和begin
for (ll a = 2;; a++) {
ll cur = a * a;
if (cur > n / a) break;
cur *= a;
while (cur <= n) {
//*phi
ans = (ans + n / cur % mod * phi[a]) % mod;
if (cur > n / a) break;
cur *= a;
}
}
//暴力end
return ans;
}
int main() {
int T;
init();
scanf("%d", &T);
while (T--) {
ll n;
scanf("%lld", &n);
printf("%lld\n", solve(n));
}
return 0;
}