洛古 P2568 莫比乌斯

#include<bits/stdc++.h>
#define LL long long
using namespace std;
const int maxn=1e7+10;
bool vis[maxn];
int prime[maxn];
int mu[maxn];
int sum[maxn];
int cnt=0;
void get_mu()// mo bi su si han shu
{
    mu[1]=1;  vis[1]=1;
    for(int i=2;i<maxn;i++)
    {
        if(!vis[i]){ prime[++cnt]=i; mu[i]=-1;}
        for(int j=1;j<=cnt&& prime[j]*i<maxn;j++)
        {
            vis[prime[j]*i]=1;
            if(i%prime[j]==0)break;
            mu[i*prime[j]]=-mu[i];
        }
    }
}
int main()
{
    get_mu();
    int n; cin>>n;
    LL ans=0;
    for(int i=1;i<=cnt;i++)
    {
        for(int j=prime[i];j<=n;j+=prime[i])
        {
            ans+=mu[j/prime[i]]*1LL*(n/j)*(n/j);
        }
    }
    cout<<ans<<endl;
}