BZOJ_P2820 YY的GCD(数论+莫比乌斯反演)

BZOJ传送门
Time Limit: 10 Sec Memory Limit: 512 MB
Submit: 1227 Solved: 636
[Submit][Status][Discuss]
Description
神犇YY虐完数论后给傻×kAc出了一题
给定N, M,求1<=x<=N, 1<=y<=M且gcd(x, y)为质数的(x, y)有多少对
kAc这种傻×必然不会了,于是向你来请教……
多组输入

Input
第一行一个整数T 表述数据组数
接下来T行,每行两个正整数,表示N, M

Output
T行,每行一个整数表示第i组数据的结果

Sample Input
2
10 10
100 100

Sample Output
30
2791

HINT
T = 10000
N, M <= 10000000

Source

直接枚举质数会超时,心情不好不想写题解了QuQ

#include
#include
using namespace std;
#define N 10000005
typedef long long ll;
int n,m,cnt,t;int p[N],mu[N];
ll g[N],sum[N];ll ans;bool b[N];
inline int in(int x=0,char ch=getchar(),int v=1){
    while(ch!='-'&&(ch>'9'||ch<'0')) ch=getchar();if(ch=='-') v=-1,ch=getchar();
    while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();return x*v;}
void GetMu(int n){
    mu[1]=1;
    for(int i=2;i<=n;i++){
        if(!b[i]) mu[i]=-1,p[cnt++]=i,g[i]=1;
        for(int j=0;j1;
            if(i%p[j]) mu[i*p[j]]=-mu[i],g[i*p[j]]=mu[i]-g[i];
            else{g[i*p[j]]=mu[i];break;}
        }
    }
    for(int i=1;i<=n;i++) sum[i]=sum[i-1]+g[i];
}
int main(){
    GetMu(N-1);t=in();
    while(t--){
        n=in(),m=in();ans=0;if(n>m) swap(n,m);
        for(int i=1,j;i<=n;i=j+1){
            j=min(n/(n/i),m/(m/i));
            ans+=(sum[j]-sum[i-1])*(n/i)*(m/i);
        }
        printf("%lld\n",ans);
    }
    return 0;
}

你可能感兴趣的:(BZOJ,数论,莫比乌斯反演)