[BZOJ3994][SDOI2015]约数个数和（莫比乌斯反演）

题目描述

传送门

题解

首先考虑如何求$d(n,m)$
有一个很神的结论：

$$d(nm)=\sum\limits_{i|n}\sum\limits_{j|m}[gcd(i,j)=1]$$

题目要求

$$\sum\limits_{i=1}^N\sum\limits_{j=1}^M d(ij)$$
令 $N<M$
可以知道上面那个式子的前缀和即为答案，即
$$\sum\limits_{i=1}^N\sum\limits_{j=1}^M d(ij)=\sum\limits_{n=1}^N\sum\limits_{m=1}^M\sum\limits_{i|n}\sum\limits_{j|m}[gcd(i,j)=1]$$
那么再化简一下
$$\sum\limits_{i=1}^N\sum\limits_{j=1}^M d(ij)=\sum\limits_{n=1}^N\sum\limits_{i|n}\sum\limits_{m=1}^M\sum\limits_{j|m}\sum\limits_{d=1}^N[d|i][d|j]\mu(d)$$
观察这个式子，选出来的i其实是[1,N]范围内所有数的约数，会有很多重复的，重复的个数为 $\lfloor {N\over i}\rfloor$
那么改一下形式
$$\sum\limits_{i=1}^N\sum\limits_{j=1}^M d(ij)=\sum\limits_{i=1}^N\sum\limits_{j=1}^M\lfloor{N\over i}\rfloor\lfloor{M\over j}\rfloor\sum\limits_{d=1}^N[d|i][d|j]\mu(d)$$
令i=id j=jd
$$\sum\limits_{i=1}^N\sum\limits_{j=1}^M d(ij)=\sum\limits_{d=1}^N \mu(d)\sum\limits_{i=1}^{N\over d}\sum\limits_{j=1}^{M\over d}\lfloor{N\over {id}}\rfloor\lfloor{M\over {jd}}\rfloor$$
$$\sum\limits_{i=1}^N\sum\limits_{j=1}^M d(ij)=\sum\limits_{d=1}^N \mu(d)\sum\limits_{i=1}^{N\over d}\sum\limits_{j=1}^{M\over d}\lfloor{{N\over d}\over i}\rfloor\lfloor{{M\over d}\over j}\rfloor$$
设 $f(n)=\sum\limits_{i=1}^n \lfloor{n\over i}\rfloor$
那么 $$\sum\limits_{i=1}^N\sum\limits_{j=1}^M d(ij)=\sum\limits_{d=1}^N \mu(d)f({N\over d})f({M\over d})$$

其中f用分块可以在$O(n\sqrt n)$时间内求出
$\mu$可以$O(n)$筛出来
但是这道题有多组数据直接暴力枚举d还是不可做。
再观察，$f({N\over d})$$f({M\over d})$只有$\sqrt n+\sqrt m$种取值，也可以分块来求。
时间复杂度$O(n\sqrt n+T*(\sqrt n+\sqrt m))$

代码

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define LL long long
const int N=5e4;
int mu[N+5],p[N+5],prime[N+5],f[N+5];
int T,n,m,p1,p2;
LL ans;

inline void get_mu(){
    mu[1]=1;
    for (int i=2;i<=N;++i){
        if (!p[i]){
            prime[++prime[0]]=i;
            mu[i]=-1;
        }
        for (int j=1;j<=prime[0]&&i*prime[j]<=N;++j){
            p[i*prime[j]]=1;
            if (i%prime[j]==0){
                mu[i*prime[j]]=0;
                break;
            }
            else mu[i*prime[j]]=-mu[i];
        }
    }
}
inline int get_f(int n){
    int ans=0;
    for (int i=1,j;i<=n;i=j+1){
        j=n/(n/i);
        ans+=(n/i)*(j-i+1);
    }
    return ans;
}

int main(){
    get_mu();
    for (int i=1;i<=N;++i) mu[i]+=mu[i-1];
    for (int i=1;i<=N;++i)
      f[i]=get_f(i);
    scanf("%d",&T);
    while (T--){
        scanf("%d%d",&n,&m); if (n>m) swap(n,m);
        ans=0; p1=p2=1;
        for (int i=1,j;i<=n;i=j+1){
            j=min(n/(n/i),m/(m/i));
            ans+=(LL)f[n/i]*f[m/i]*(mu[j]-mu[i-1]);
        }
        printf("%lld\n",ans);
    }
}