【HDU6683】Rikka with Geometric Sequence（杜教筛）（非常规数论分块）

传送门

---

题解：

长度为 1 和 2 的等比数列个数是个人都会。

考虑长度为3及以上的等比数列个数可以考虑公比。

设公比 $q=a/b$，其中$a>b，gcd(a,b)=1$，设长度为$k$，则数列头显然有 $\lfloor \frac{n}{a^{k-1}}\rfloor$ 中选择，合法的 $b$ 有 $\varphi (a)$ 种选择。

那么答案为 $$\sum _{i=2}^n\varphi (i)\sum _{k=2}\lfloor \frac{n}{i^k}\rfloor$$

这样 $i$ 需要考虑到 $\sqrt{n}$，无法接受。

把长度为 $3$ 的等比数列提出来单独算，答案为$\sum _{i=2}^n\varphi (i)\lfloor \frac{n}{i^2}\rfloor$

对 $i^2$ 数论分块，分段只有 $O(\sqrt[3]{n})$ 段

剩下的暴力枚举到 $\sqrt[3]{n}$ 即可。

复杂度玄学。

---

代码：

#include<bits/stdc++.h>
#define ll long long
#define re register
#define cs const

using std::cerr;
using std::cout;

cs int mod=998244353;
inline int add(int a,int b){return a+b>=mod?a+b-mod:a+b;}
inline int dec(int a,int b){return a-b<0?a-b+mod:a-b;}
inline int mul(int a,int b){ll r=(ll)a*b;return r>=mod?r%mod:r;}
inline void Inc(int &a,int b){a+=b-mod;a+=a>>31&mod;}
inline void Dec(int &a,int b){a-=b;a+=a>>31&mod;}
inline void Mul(int &a,int b){a=mul(a,b);}
inline int MD(ll x){return x>=mod?x%mod:x;}

cs int N=5e7+7;

int p[N],pc;
int phi[N],sum_phi[N];
bool mrk[N];

void linear_sieves(){
	phi[1]=1;sum_phi[1]=1;
	for(int re i=2;i<N;++i){
		if(!mrk[i])p[++pc]=i,phi[i]=i-1;
		for(int re j=1;p[j]*i<N;++j){
			mrk[i*p[j]]=true;
			if(i%p[j]==0){
				phi[i*p[j]]=phi[i]*p[j];
				break;
			}phi[i*p[j]]=phi[i]*(p[j]-1);
		}
		sum_phi[i]=add(sum_phi[i-1],phi[i]);
	}
}

std::unordered_map<ll,int> Map;

inline int Sum(ll x){
	x=MD(x);return MD(x*(x+1)>>1); 
}
int S(ll n){
	if(n<N)return sum_phi[n];
	if(Map.find(n)!=Map.end())return Map[n];
	int &res=Map[n]=Sum(n);
	for(ll l=2,r,u;(u=n/l);l=r+1)
		Dec(res,mul(MD((r=n/u)-l+1),S(u)));
	return res;
}

ll n;

void Main(){
	linear_sieves();
	int T;scanf("%d",&T);
	while(T--){
		scanf("%lld",&n);
		int ans=Sum(n);
		for(ll l=2,r,u;(u=n/l/l);l=r+1)
			Inc(ans,mul(dec(S(r=sqrt(n/u)),S(l-1)),MD(u)));
		int lim=pow(n,1./3.);
		for(int re i=2;i<=lim;++i)
			for(ll x=(ll)i*i*i,li=n/i;;x*=i){
				Inc(ans,mul(phi[i],MD(n/x)));
				if(x>li)break;
			}
		cout<<ans<<"\n";
	}
}

inline void file(){
#ifdef zxyoi
	freopen("seq.in","r",stdin);
#endif
}
signed main(){file();Main();return 0;}