洛谷P2257 YY的GCD

洛谷P2257 YY的GCD

题目大意：求有多少数对$(x,y)$$(1<=x<=n,1<=y<=m)$满足$gcd(x,y)$为质数。

此题需要用到莫比乌斯反演。

题意即求

$\sum _{k=1}^n\sum _{i=1}^n\sum _{j=1}^m[gcd(i,j)==k](k\in prime)$

两边同时除以$k$，将$[gcd(i,j)==k]$化成$[gcd(i,j)==1]$

$\sum _{k=1}^n\sum _{i=1}^{\lfloor \frac{n}{k}\rfloor }\sum _{j=1}^{\lfloor \frac{m}{k}\rfloor }[gcd(i,j)==1](k\in prime)$

我们知道莫比乌斯函数的性质

$\sum _{d|n}\mu (d)=[n==1]$

我们把$n$换成$gcd(i,j)$，得$[gcd(i,j)==1]=\sum _{d|gcd(i,j)}\mu (d)$

原式变为

$\sum _{k=1}^n\sum _{i=1}^{\lfloor \frac{n}{k}\rfloor }\sum _{j=1}^{\lfloor \frac{m}{k}\rfloor }\sum _{d|gcd(i,j)}\mu (d)(k\in prime)$

枚举$d$

$\sum _{k=1}^n\sum _{d=1}^{\lfloor \frac{n}{k}\rfloor }\mu (d)\ast \lfloor \frac{n}{kd}\rfloor \ast \lfloor \frac{m}{kd}\rfloor (k\in prime)$

令$T=kd$，有

$\sum _{k=1}^n\sum _{d=1}^{\lfloor \frac{n}{k}\rfloor }\mu (d)\ast \lfloor \frac{n}{T}\rfloor \ast \lfloor \frac{m}{T}\rfloor (k\in prime)$

枚举$T$，提到前面

$\sum _{T=1}^n\lfloor \frac{n}{T}\rfloor \ast \lfloor \frac{m}{T}\rfloor \sum _{k|T,k\in prime}\mu (\frac{T}{k})$

用数论分块处理即可，时间复杂度为$O(n\ln n+T\sqrt{n})$。

code

#include<bits/stdc++.h>
#define N 10000000
using namespace std;
int t,n,m,fl[N+5],p[N+5],mu[N+5],f[N+5];
long long ans,sum[N+5];
void dd(){
    mu[1]=1;
    for(int i=2;i<=N;i++){
        if(!fl[i]){
            p[++p[0]]=i;mu[i]=-1;
        }
        for(int j=1;j<=p[0]&&i*p[j]<=N;j++){
            fl[i*p[j]]=1;
            if(i%p[j]==0){
            	mu[i*p[j]]=0;
            	break;
            }
            mu[i*p[j]]=-mu[i];
        }
    }
    for(int i=1;i<=p[0];i++){
        for(int j=1;p[i]*j<=N;j++){
            f[p[i]*j]+=mu[j];
        }
    }
    for(int i=1;i<=N;i++){
        sum[i]=sum[i-1]+f[i];
    }
}
void ss(){
    ans=0;
    for(int l=1,r=0;l<=n;l=r+1){
        r=min(n/(n/l),m/(m/l));
        ans+=(sum[r]-sum[l-1])*(n/l)*(m/l);
    }
}
int main()
{
    dd();
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&m);
        if(n>m) swap(n,m);
        ss();
        printf("%lld\n",ans);
    }
    return 0;
}