Description
设函数f(d)表示d的质因数个数,f(1)=0
求
∑i=1n2f(i)
Solution
显然,式子可以写成这样:
∑i=1n∑d|i[gcd(d,id)=1]
(大家感受一下)
再变:
∑d=1n∑i=1⌊nd⌋[gcd(d,i)=1]
无中生有一发:
∑d=1n∑i=1⌊nd⌋∑j|gcd(d,i)μ(j)
交换主体:
(其实理解一下意义可以直接写出)
∑d=1nμ(d)∑i=1⌊nd2⌋⌊nd2i⌋
发现,d远远没有到达n,所以:
∑d=1n√μ(d)∑i=1⌊nd2⌋⌊nd2i⌋
复杂度:O(n−−√ln(n−−√))
Code
#include <cstdio>
#include <cstdlib>
#include <cmath>
#define fo(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
typedef long long LL;
const int N=1000500,mo=998244353;
LL n,ans;
int m;
bool prz[N];
int pr[N/5];
int mu[N];
int gcd(int x,int y){return y?gcd(y,x%y):x;}
LL FK(LL n)
{
LL ans=0;
for(LL i=1,nx;i<=n;i=nx+1)
{
nx=n/(n/i);
ans=(ans+(nx-i+1)%mo*(n/i))%mo;
}
return ans;
}
int main()
{
int q,w;
scanf("%lld",&n);
m=sqrt(n);
mu[1]=1;
fo(i,2,m+1)
{
if(!prz[i])pr[++pr[0]]=i,mu[i]=-1;
fo(j,1,pr[0])
{
int t=pr[j]*i;
if(t>m)break;
prz[t]=1;
if(i%pr[j]==0)break;
mu[t]=-mu[i];
}
}
fo(i,2,m)mu[i]=(mu[i]+mu[i-1])%mo;
ans=0;
for(LL i=1,nx;i<=m;i=nx+1)
{
nx=n/(n/(i*i));
nx=sqrt(nx);
ans=(ans+(mu[nx]-mu[i-1])%mo*FK(n/(i*i)))%mo;
}
printf("%lld\n",(ans+mo)%mo);
return 0;
}