洛谷P3455 [POI2007]ZAP-Queries：莫比乌斯反演入门题

原题链接：洛谷P3455 [POI2007]ZAP-Queries

题目描述

FGD正在破解一段密码，他需要回答很多类似的问题：对于给定的整数a,b和d，有多少正整数对$x,y$，满足$x\le a，y\le b$，并且$gcd(x,y)=d$。FGD希望得到你的帮助。

输入格式

第一行一个整数 $n(1\le n\le 50000$)，代表询问个数。
接下来$n$行每行三个整数: a, b, d$(1\le d\le a,b\le 50000)$, 用空格分开。

输出格式

共$n$行，每行一个整数，代表每次询问的答案。

输入输出样例

• 输入1

2
4 5 2
6 4 3

• 输出1

3
2

题解

此题为莫比乌斯反演入门题，让我们对莫比乌斯反演题是要我们做什么有个概念。有关莫比乌斯反演，可参考莫比乌斯反演及的证明与应用

我们设：
$$f(x)=\sum _{i=1}^a\sum _{j=1}^b[gcd(i,j)=x],$$
$$F(x)=\sum _{x|k}f(k)=\sum _{i=1}^a\sum _{j=1}^b[x|gcd(i,j)]$$
$$=\sum _{i=1}^a[x|i]\sum _{j=1}^b[x|j]=\lfloor \frac{a}{x}\rfloor \lfloor \frac{b}{x}\rfloor$$
则可以由莫比乌斯反演可以推出：
$$f(x)=\sum _{x|k}\mu (\lfloor \frac{k}{x}\rfloor )F(k)$$
设完这两个函数之后，我们便发现，$Ans=f(d)$。那么，
$$Ans=\sum _{d|k}\mu (\lfloor \frac{k}{d}\rfloor )F(k)$$
设$t=\frac{k}{d}$。我们枚举$t$：
$$Ans=\sum _{t=1}\mu (t)\lfloor \frac{a}{td}\rfloor \lfloor \frac{b}{td}\rfloor$$
由于有多组数据，所以我们再用一下整除分块预处理，这样单次询问就可以$O(\sqrt{n})$了。要注意的是，当$td>min(a,b)$时就可以结束累加。

当然还有最后一个问题：$\lfloor \frac{a}{td}\rfloor$含有一个常量$d$，如何整除分块？
设$a=kt+b$ $(0\le b<t)$, $k=pd+q$ $(0\le q<d)$, 那$$\lfloor \frac{\lfloor \frac{a}{t}\rfloor }{d}\rfloor =\lfloor \frac{k}{d}\rfloor =p$$
$$\lfloor \frac{a}{td}\rfloor =\lfloor \frac{(pd+q)t+b}{td}\rfloor$$
$$=\lfloor \frac{ptd+(tq+b)}{td}\rfloor$$
$$p+\lfloor \frac{tq+b}{td}\rfloor =p$$
因此$$\lfloor \frac{a}{td}\rfloor =\lfloor \frac{\lfloor \frac{a}{t}\rfloor }{d}\rfloor =\lfloor \frac{\lfloor \frac{a}{d}\rfloor }{t}\rfloor ,$$
整除分块时块与块之间的距离不小于$\lfloor \frac{a}{t}\rfloor$。当然，我们也可以先将$a,b$分别除以$d$之后再进行分块。

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn=50005;
int n,a,b,d;

int num[maxn],prime[maxn],p_num;
int mu[maxn];
int sum[maxn];
void get_mu()
{
    mu[1]=1;
    for(int i=2;i<maxn;i++)
    {
        if(!num[i])
        {
            prime[++p_num]=i;
            mu[i]=-1;
        }
        for(int j=1;j<=p_num&&i*prime[j]<maxn;j++)
        {
            num[i*prime[j]]=1;
            if(i%prime[j]) mu[i*prime[j]]=-mu[i];
            else
            {
                mu[i*prime[j]]=0;
                break;
            }
        }
    }
    //预处理mu的前缀和。
    for(int i=1;i<maxn;i++) sum[i]=sum[i-1]+mu[i];
}


int main()
{
    scanf("%d",&n);
    get_mu();
    while(n--)
    {
        int ans=0;
        scanf("%d%d%d",&a,&b,&d);
        int up=min(a,b);
        //数论分块
        //另一种方法：
        //int ss=a/d,sss=b/d;
        for(int i=1,j;i*d<=up;i=j+1)
        {
            j=min(a/(a/i),b/(b/i));
            //另一种方法：
        	//j=min(ss/(ss/i),sss/(sss/i));
            ans+=(a/(i*d))*(b/(i*d))*(sum[j]-sum[i-1]);
        }
        printf("%d\n",ans);
    }
    return 0;
}