bzoj3994: [SDOI2015]约数个数和

链接

　　http://www.lydsy.com/JudgeOnline/problem.php?id=3994

这道题目学到的

　　数学是硬伤….OI早晚要受到牵连，这不，叫这道题卡了一晚上。
　　有些方法经常要用到
　　一、枚举次序的变化，如果要先枚举一个数字$x$，再枚举其倍数$d$，可以把$d$拿到前面来，先枚举$d$再枚举其因子$x$，这样的好处是$x$的上界变成了$d$，而不是和询问有关的$n$。
　　二、$\sum_{i=1}^n\sum_{j=1}^MA[gcd(i,j)=1]$其中A是随便一个式子，就等价于$\sum_{i=1}^n\sum_{j=1}^MA\sum_{d|gcd(i,j)}\mu(d)$
　　三、$\lfloor\frac{n}{id}\rfloor=\lfloor\frac{\lfloor\frac{n}{d}\rfloor}{i}\rfloor$
　　四、$\sum_i\sum_jf(i)g(j)=\sum_if(i)\sum_jg(j)$，就是说$\sum$后面和枚举变量没关的东西统统可以拿出来。

题解

　　一个结论，

$$\sum_{i=1}^N\sum_{j=1}^Md(ij)=\sum_{i-1}^N\sum_{j=1}^M\lfloor\frac{N}{i}\rfloor\lfloor\frac{M}{j}\rfloor[gcd(i,j)=1]$$
　　应用第二条
　　 $$=\sum_{i=1}^N\sum_{j=1}^M\lfloor\frac{N}{i}\rfloor\lfloor\frac{M}{j}\rfloor\sum_{d|gcd(i,j)}\mu(d)$$
　　应用第一、四条
　　 $$=\sum_{d=1}^{min(N,M)}\mu(d)\sum_{d|i}^N\lfloor\frac{N}{i}\rfloor\sum_{d|j}^M\lfloor\frac{M}{j}\rfloor$$
　　继续
　　 $$=\sum_{d=1}^{min(N,M)}\mu(d) \sum_{i=1}^{\lfloor\frac{N}{d}\rfloor} \lfloor\frac{N}{id}\rfloor \sum_{j=1}^{ \lfloor\frac{M}{d}\rfloor} \lfloor\frac{M}{jd}\rfloor$$
　　应用第三条
　　 $$=\sum_{d=1}^{min(N,M)}\mu(d) \sum_{i=1}^{\lfloor\frac{N}{d}\rfloor} \lfloor\frac{\lfloor\frac{N}{d}\rfloor}{i}\rfloor \sum_{j=1}^{ \lfloor\frac{M}{d}\rfloor} \lfloor\frac{\lfloor\frac{M}{d}\rfloor}{j}\rfloor$$
　　可以发现，后面两个 $\sum$的值都只和上界有关，因此可以预处理 $f[x]=\sum_{i=1}^x\lfloor\frac{x}{i}\rfloor$
　　最终的答案为
　　 $$\sum_{d=1}^{min(n,m)}\mu(d) f(\lfloor\frac{N}{d}\rfloor) f(\lfloor\frac{M}{d}\rfloor)$$
　　这个可以 $\sqrt N$做
　　 $f(x)$的预处理，我最开始是用 $N\sqrt N$处理的，后来听说这个东西就是约数个数和，于是改用了 $NlogN$的方法。
　　代码在bzoj是AC的，但是洛谷上会TLE（我怎么知道是什么情况？？）。

代码

//莫比乌斯反演 
#include <cstdio>
#include <algorithm>
#define ll long long
#define maxn 50000
using namespace std;
ll mu[maxn+10], prime[maxn+10], f[maxn+10], mark[maxn+10], s[maxn+10];
void init()
{
    ll i, j;
    s[1]=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];
        }
        s[i]=s[i-1]+mu[i];
    }
    for(i=1;i<=maxn;i++)
        for(j=i;j<=maxn;j+=i)f[j]++;
    for(i=1;i<=maxn;i++)f[i]+=f[i-1];
}
ll calc(ll n, ll m)
{
    ll ans=0, i, last;
    if(n>m)swap(n,m);
    for(i=1;i<=n;i=last+1)
    {
        last=min(n/(n/i),m/(m/i));
        ans+=(s[last]-s[i-1])*f[n/i]*f[m/i];
    }
    return ans;
}
int main()
{
    ll T, N, M;
    init();
    scanf("%lld",&T);
    while(T--)
    {
        scanf("%lld%lld",&N,&M);
        printf("%lld\n",calc(N,M));
    }
    return 0;
}