【BZOJ2301】【HAOI2011】Problem b 莫比乌斯反演

Mission

对于给出的n个询问，每次求有多少个数对(x,y)，满足a≤x≤b，c≤y≤d，且gcd(x,y) = k，gcd(x,y)函数为x和y的最大公约数。

1≤n≤50000，1≤a≤b≤50000，1≤c≤d≤50000，1≤k≤50000

Solution

裸的莫比乌斯反演。

把询问拆分成四个子询问，然后莫比乌斯反演要用分块求解。

Code

#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<math.h>
#include<string.h>
#define ll long long
using namespace std;
const char* fin="ex2301.in";
const char* fout="ex2301.out";
const ll inf=0x7fffffff;
const ll maxn=50007;
ll t,a,b,c,d,ind,i,j,k;
ll miu[maxn],p[maxn];
bool bz[maxn];
ll ans;
ll f(ll n,ll m){
ll i,j,k,ans=0;
if (n<=0 || m<=0) return 0;
if (n>m) swap(n,m);
for (i=1;ind*i<=n;){
j=min(n/(ind*(n/(ind*i))),m/(ind*m/(ind*i)));
ans+=(n/(ind*i))*(m/(ind*i))*(miu[j]-miu[i-1]);
i=j+1;
}
return ans;
}
int main(){
freopen(fin,"r",stdin);
freopen(fout,"w",stdout);
scanf("%lld",&t);
miu[1]=1;
for (i=2;i<maxn;i++){
if (!bz[i]){
p[++p[0]]=i;
miu[i]=-1;
}
for (j=1;j<=p[0];j++){
k=i*p[j];
if (k>=maxn) break;
bz[k]=true;
if (i%p[j]==0){
miu[k]=0;
break;
}else miu[k]=-miu[i];
}
}
for (i=1;i<maxn;i++) miu[i]+=miu[i-1];
while (t--){
scanf("%lld%lld%lld%lld%lld",&a,&b,&c,&d,&ind);
printf("%lld\n",f(b,d)-f(b,c-1)-f(a-1,d)+f(a-1,c-1));
}
return 0;
}