BZOJ3994 || P3327 [SDOI2015]约数个数和【莫比乌斯反演】

Time Limit: 20 Sec
Memory Limit: 128 MB

Description

设d(x)为x的约数个数，给定N、M，求${\sum }_{i=1}^n{\sum }_{j=1}^{m}d(ij)$

Input

输入文件包含多组测试数据。

第一行，一个整数T，表示测试数据的组数。
接下来的T行，每行两个整数N、M。

Output

T行，每行一个整数，表示你所求的答案。

HINT

1<=N, M<=50000
1<=T<=50000

---

题目分析

在蒟蒻望着屏幕发呆对此题无从下手半个小时后
点开题解发现上来就是一个结论之时内心是无比复杂的，复杂到已经语无伦次。。。

先看看这个蒟蒻不懂怎么证得结论
$$d(ij)=\sum _{x|i}\sum _{y|j}[gcd(x,y)=1]$$
有了这个结论一切都好办了，直接带入
$$\sum _{i=1}^n\sum _{j=1}^m\sum _{x|i}\sum _{y|j}[gcd(x,y)=1]$$
再莫比乌斯反演
$$\sum _{i=1}^n\sum _{j=1}^m\sum _{x|i}\sum _{y|j}\sum _{d|gcd(x,y)}\mu (d)$$
把枚举$i,j$的约数变成枚举$x,y$的倍数
$$\sum _{i=1}^n\sum _{j=1}^m\sum _{x=1}^i\sum _{y=1}^j[x|i][y|j]\sum _{d=1}^{gcd(x,y)}\mu (d)[d|gcd(x,y)]$$

$$\sum _{x=1}^n\sum _{y=1}^m\sum _{t_1=1}^{\lfloor \frac{n}{x}\rfloor }\sum _{t_2=1}^{\lfloor \frac{m}{y}\rfloor }\sum _{d=1}^{gcd(x,y)}\mu (d)[d|gcd(x,y)]$$
把d的枚举提前，把枚举$x,y$变成枚举$d$的倍数
$$\sum _{d=1}^{min(n,m)}\sum _{x=1}^n\sum _{y=1}^m\sum _{t_1=1}^{\lfloor \frac{n}{x}\rfloor }\sum _{t_2=1}^{\lfloor \frac{m}{y}\rfloor }\mu (d)[d|gcd(x,y)]$$

$$\sum _{d=1}^{min(n,m)}\sum _{k_1=1}^{\lfloor \frac{n}{d}\rfloor }\sum _{k_2=1}^{\lfloor \frac{m}{d}\rfloor }\sum _{t_1=1}^{\lfloor \frac{n}{d{k}_1}\rfloor }\sum _{t_2=1}^{\lfloor \frac{m}{d{k}_2}\rfloor }\mu (d)=\sum _{d=1}^{min(n,m)}\mu (d)\sum _{k_1=1}^{\lfloor \frac{n}{d}\rfloor }\sum _{k_2=1}^{\lfloor \frac{m}{d}\rfloor }\sum _{t_1=1}^{\lfloor \frac{n}{d{k}_1}\rfloor }\sum _{t_2=1}^{\lfloor \frac{m}{d{k}_2}\rfloor }1$$

看起来不太简洁啊，令$T_1=\lfloor \frac{n}{d}\rfloor ,T_2=\lfloor \frac{m}{d}\rfloor$

$$\sum _{d=1}^{min(n,m)}\mu (d)\sum _{i=1}^{T_1}\sum _{j=1}^{T_2}\sum _{x=1}^{\lfloor \frac{T_1}{i}\rfloor }\sum _{y=1}^{\lfloor \frac{T_2}{j}\rfloor }1=\sum _{d=1}^{min(n,m)}\mu (d)\sum _{i=1}^{T_1}\lfloor \frac{T_1}{i}\rfloor \sum _{j=1}^{T_2}\lfloor \frac{T_2}{j}\rfloor$$

预处理$\mu (i)$的前缀和以及$g(n)={\sum }_{i=1}^n\lfloor \frac{n}{i}\rfloor$之后整除分块求解答案即可

这里$g(n)$预处理复杂度是$O(n\sqrt{n})$，其实是还有优化的余地的
仔细观察可以发现$g(n)$其实就是${\sum }_{i=1}^{n}d(i)$(其中d(i)为i的约数个数)
由于约数个数是积性函数，完全可以与莫比乌斯函数一起处理
(然而因为蒟蒻比较懒就没写了)

---

#include<iostream>
#include<vector>
#include<algorithm>
#include<queue>
#include<cstring>
#include<cstdio>
using namespace std;
typedef long long lt;
 
int read()
{
    int f=1,x=0;
    char ss=getchar();
    while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
    while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
    return f*x;
}

const int maxn=50010;
int T;
int miu[maxn];
int vis[maxn],prim[maxn],cnt;
lt sum[maxn],ds[maxn];

void Miu(int n)
{
    miu[1]=1;
    for(int i=2;i<=n;++i)
    {
        if(!vis[i]) prim[++cnt]=i,miu[i]=-1;
        for(int j=1;j<=cnt;++j)
        {
            if(i*prim[j]>n) break;
            vis[i*prim[j]]=1;
            if(i%prim[j]==0) break;
            else miu[i*prim[j]]=-miu[i];
        }
    }
    for(int i=1;i<=n;++i)
    sum[i]=sum[i-1]+miu[i];
    
    for(int i=1;i<=n;++i)
    for(int ll=1,rr;ll<=i;ll=rr+1)
    {
        rr=i/(i/ll);
        ds[i]+=1ll*(i/ll)*(rr-ll+1);
    }
}

lt query(int n,int m)
{
    lt res=0; int lim=min(n,m);
    for(int ll=1,rr;ll<=lim;ll=rr+1)
    {
        rr=min(n/(n/ll),m/(m/ll));
        res+=1ll*(sum[rr]-sum[ll-1])*ds[n/ll]*ds[m/ll];
    }
    return res;
}

int main()
{
    T=read(); Miu(maxn-5);
    while(T--)
    {
        int n=read(),m=read();
        printf("%lld\n",query(n,m));
    }
    return 0;
}