1. 程式人生 > >Bzoj 2301: [HAOI2011]Problem b(莫比烏斯反演+除法分塊)

Bzoj 2301: [HAOI2011]Problem b(莫比烏斯反演+除法分塊)

2301: [HAOI2011]Problem b
Time Limit: 50 Sec Memory Limit: 256 MB
Description
對於給出的n個詢問,每次求有多少個數對(x,y),滿足a≤x≤b,c≤y≤d,且gcd(x,y) = k,gcd(x,y)函式為x和y的最大公約數。
Input
第一行一個整數n,接下來n行每行五個整數,分別表示a、b、c、d、k
Output
共n行,每行一個整數表示滿足要求的數對(x,y)的個數
Sample Input
2
2 5 1 5 1
1 5 1 5 2
Sample Output
14
3
HINT
100%的資料滿足:1≤n≤50000,1≤a≤b≤50000,1≤c≤d≤50000,1≤k≤50000

/*
莫比烏斯反演.
好吧這題比上一題簡單.
然後容斥的話用二維矩陣想一想就行了.
一開始推式子的時候把推錯了一個取值 (打手.
最後是這個東西∑(min(n/k,m/k),d=1)mu[d]*[n/kd][m/kd]. 
樸素是O(n/k)的,用除法分塊優化以後可以降到O(2√n).
用cout輸出BZOJ判 Wrong 不知道為啥.
*/
#include<iostream>
#include<cstdio>
#define MAXN 50001
#define LL long long
using namespace std;
int t,a,b,c,d,k,tot,last,mu[MAXN],pri[MAXN];
LL ans,sum[MAXN];
bool vis[MAXN];
void pre()
{
    mu[1
]=1; for(int i=2;i<=MAXN-1;i++) { if(!vis[i]) vis[i]=true,pri[++tot]=i,mu[i]=-1; for(int j=1;j<=tot&&i*pri[j]<=MAXN-1;j++) { vis[i*pri[j]]=true; if(i%pri[j]) mu[i*pri[j]]=-mu[i]; else {mu[i*pri[j]]=0;break;} } } for
(int i=1;i<=MAXN-1;i++) sum[i]=sum[i-1]+mu[i]; } LL slove(LL n,LL m) { ans=0;n/=k,m/=k; for(LL i=1;i<=min(n,m);i=last+1) { last=min(n/(n/i),m/(m/i)); ans+=(n/i)*(m/i)*(sum[last]-sum[i-1]); } return ans; } int main() { pre(); scanf("%d",&t); while(t--) { scanf("%d%d%d%d%d",&a,&b,&c,&d,&k); printf("%lld\n",slove(b,d)-slove(b,c-1)-slove(a-1,d)+slove(a-1,c-1)); } return 0; }