BZOJ2301: [HAOI2011]Problem b（洛谷P2522）

莫比乌斯反演

BZOJ题目传送门
洛谷题目传送门

首先套路容斥，变成求$gcd(x,y)=n$的$(x,y)$个数$(1\leq x\leq n,1\leq y \leq m)$

设$f(i)$为$gcd(x,y)=i$的$(x,y)$个数，$F(i)$为$i|gcd(x,y)$的$(x,y)$个数$(1\leq x\leq n,1\leq y \leq m)$

很快可以得到$F(i)=\lfloor\frac ni\rfloor\lfloor\frac mi\rfloor$

然后反演一下：$f(i)=\sum_{i|d}\mu(\frac di)F(n)=\sum_{i|d}\mu(\frac di)\lfloor\frac nd\rfloor\lfloor\frac md\rfloor$

我们把原来的$n,m$同时除以$k$，这样其实就是求$f(1)=\sum\mu(d)\lfloor\frac nd\rfloor\lfloor\frac md\rfloor$

然后预处理$\mu(d)$的前缀和。因为$\lfloor\frac nd\rfloor$的连续一段贡献是相同的，所以分段统计即可（貌似这种东西叫除法分块）

代码：

#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 50005
#define F inline
using namespace std;
int n,a,b,c,d,k,t,nd,p[N],mu[N],sum[N];
bool f[N];
F char readc(){
    static char buf[100000],*l=buf,*r=buf;
    if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
    if (l==r) return EOF; return *l++;
}
F int _read(){
    int x=0; char ch=readc();
    while (!isdigit(ch)) ch=readc();
    while (isdigit(ch)) x=(x<<3)+(x<<1)+(ch^48),ch=readc();
    return x;
}
F void mkp(){
    for (int i=2;i<N;i++){
        if (!f[i]) p[++nd]=i,mu[i]=-1;
        for (int j=1,v;j<=nd&&(v=i*p[j])<N;j++){
            f[v]=true,mu[v]=-mu[i];
            if (i%p[j]==0) { mu[v]=0; break; }
        }
    }
    for (int i=1;i<N;i++) sum[i]=sum[i-1]+mu[i];
}
F int calc(int n,int m){
    if (n>m) swap(n,m); int ans=0,p;
    for (int i=1;i<=n;i=p+1){
        p=min(n/(n/i),m/(m/i));
        ans+=(sum[p]-sum[i-1])*(n/i)*(m/i);
    }
    return ans;
}
int main(){
    t=_read(),mu[1]=1,mkp();
    while (t--){
        a=_read(),b=_read(),c=_read(),d=_read(),k=_read();
        a--,c--,a/=k,b/=k,c/=k,d/=k;
        int ans=calc(b,d)+calc(a,c)-calc(a,d)-calc(b,c);
        printf("%d\n",ans);
    }
    return 0;
}