洛谷P3455 [POI2007]ZAP-Queries(莫比乌斯反演学习)

链接

　　https://www.luogu.org/problem/show?pid=3455

莫比乌斯反演

　　所谓反演，就是类似于逆运算之类的一种东西。
　　像加的逆运算就是减，比如你知道$a+b=c$，又知道了$b$和$c$的值，欲求$a$，就必须引入一种新的运算”-“。像乘法逆元啊，什么的，作用大抵都是如此。
　　对于一种递推式：

$$F(n)=\sum_{d|n}f(d)$$
　　那么

$$f(n)=\sum_{d|n}\mu(d)F(\frac{n}{d})$$
　　还有一种形式： $$F(n)=\sum_{n|d}f(d)$$
　　推出 $$f(n)=\sum_{n|d}\mu(\frac{d}{n})F(d)$$
　　其中 $\mu(x)被定义为$
　　 $$\mu(x)=\begin{cases} 　　1&x=1\\(-1)^k&x=p_1p_2p_3...p_k\\ 　　0&else \end{cases}$$

题解

　　这是一个经典的莫比乌斯反演问题。
　　设$f(i)$表示$x\in[1,N],y\in[1,M]$且$gcd(x,y)=i$的数对个数。$F(i)$表示范围内$gcd(x,y)$是$i$的倍数的数对个数。一个数对要满足它们的$gcd$是$i$的倍数，则$x$和$y$中都必须包含$i$这个因子，所以$F(i)=\lfloor\frac{N}{i}\rfloor\lfloor\frac{M}{i}\rfloor$。又根据定义

$$F(x)=\sum_{x|d}^Nf(d)$$ 反演 $$f(x)=\sum_{x|d}^N\mu(\frac{d}{x})F(d)=\sum_{x|d}^{N}\mu(\frac{d}{x})\lfloor\frac{N}{d}\rfloor\lfloor\frac{M}{d}\rfloor$$
　　用 $d$表示是$x$的几倍(注意意义变了)
　　 $$f(x)=\sum_{d=1}^{N/x}\mu(d)\lfloor\frac{N}{dx}\rfloor\lfloor\frac{M}{dx}\rfloor$$
　　随便手动模拟几组数据，会发现由于整除，会有一些神奇的事情发生：
　　 $$5/1=5\\5/2=2\\5/3=1\\5/4=1\\5/5=1$$
　　可以证明, $\lfloor\frac{N}{1}\rfloor$到 $\lfloor\frac{N}{N}\rfloor$不同数字的数量级在 $\sqrt N$
　　所以可以预处理 $\mu()$函数的前缀和，根据结合律跳着计算答案即可。计算的方法在网上很普及了，详见代码。

代码

//莫比乌斯反演
#include <cstdio> 
#include <algorithm>
#define maxn 50000
using namespace std;
int mu[maxn+10], mark[maxn+10], prime[maxn+10], s[maxn+10];
void init()
{
    int i, j;
    mu[1]=1;
    for(i=2;i<=maxn;i++)
    {
        if(!mark[i])prime[++prime[0]]=i,mu[i]=-1;
        for(j=1;j<=prime[0] and i*prime[j]<=maxn;j++)
        {
            mark[i*prime[j]]=1;
            if(i%prime[j]==0){mu[i*prime[j]]=0;break;}
            mu[i*prime[j]]=-mu[i];
        }
    }
    for(i=1;i<=maxn;i++)s[i]=s[i-1]+mu[i];
}
int calc(int n, int m)
{
    int i, last, ans=0;
    if(n>m)swap(n,m);
    for(i=1;i<=n;i=last+1)
    {
        last=min(n/(n/i),m/(m/i));
        ans+=(n/i)*(m/i)*(s[last]-s[i-1]);
    }
    return ans;
}
int main()
{
    int T, a, b, d;
    init();
    for(scanf("%d",&T);T;T--)
    {
        scanf("%d%d%d",&a,&b,&d);
        printf("%d\n",calc(a/d,b/d));
    }
    return 0;
}