【反演复习计划】【bzoj2818】gcd

就是之前的2820的升级版。

把暴力枚举素数改成预处理就随便A了。

#include<bits/stdc++.h>
#define N 10000005
#define ll long long
using namespace std;
int mu[N],vis[N],prime[N],cnt;
long long f[N];
void calcmu(){
    cnt=0;mu[1]=1;
    memset(f,0,sizeof(f));
    memset(vis,true,sizeof(vis));
    for(int i=2;i<N;i++){
        if(vis[i])prime[++cnt]=i,mu[i]=-1,f[i]=1;
        for(int j=1;j<=cnt;j++){
            int t=prime[j]*i;
            if(t>N)break;
            vis[t]=false;
            if(i%prime[j]==0){mu[t]=0;f[t]=mu[i];break;}
            mu[t]-=mu[i];f[t]=mu[i]-f[i];
        }
    }
    for(int i=1;i<=N;i++)f[i]+=f[i-1];
}
int read(){
    int f=1,x=0;char ch;
    do{ch=getchar();if(ch=='-')f=-1;}while(ch<'0'||ch>'9');
    do{x=x*10+ch-'0';ch=getchar();}while(ch>='0'&&ch<='9');
    return x*f;
}
int main(){
    calcmu();int j;int n,m;
    int T=1;
    while(T--){
        ll ans=0;n=read();m=n;
        if(n>m)swap(n,m);
        for(int i=1;i<=n;i=j+1){
            j=n/(n/i); 
            ans+=(f[j]-f[i-1])*(n/i)*(m/i);
        }
        printf("%lld\n",ans);
    }
    return 0;
}