BZOJ 3994 莫比乌斯反演

题目链接

---

题意：已知d(x)为x的约数个数，给定N,M(<=50000)，求：
$$\sum _{i=1}^n\sum _{j=1}^{m}d(ij)$$

思路：
直接利用公式：
$$d(nm)=\sum _{i|n}\sum _{j|m}[gcd(i,j)==1]$$
前缀和：
$$\sum _{i=1}^n\sum _{j=1}^{m}d(ij)=\sum _{i=1}^n\sum _{j=1}^m\lfloor \frac{n}{i}\rfloor \lfloor \frac{m}{j}\rfloor [gcd(i,j)==1]$$

证明参考博客
什么二大重数学归纳，反正我是没看懂= =
不过直接考虑质因子的贡献还是比较好想通的。

这样的话，设答案为Ans，则：
$$Ans=\sum _{i=1}^n\sum _{j=1}^m\lfloor \frac{n}{i}\rfloor \lfloor \frac{m}{j}\rfloor [gcd(i,j)==1]$$
由${\sum }_{d|n}\mu (d)=[n==1]$，得：
$$Ans=\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)$$
转换枚举变量，得：
$$Ans=\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$$

令：$f(x)={\sum }_{i=1}^x\lfloor \frac{x}{i}\rfloor$

得出：
$$Ans=\sum _{d=1}^{min(n,m)}\mu (d)f(\lfloor \frac{n}{d}\rfloor )f(\lfloor \frac{m}{d}\rfloor )$$

故$O(n\sqrt{n})$预处理$f(x)$然后分块就好啦

代码：

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
typedef long long ll;
 
const int A = 5e4 + 10;
int pri[A],tot;
bool vis[A];
ll d[A],mu[A];
 
void init(){
    tot = 0;mu[1] = 1;
    for(int i=2 ;i<A ;i++){
        if(!vis[i]){mu[i] = -1;pri[++tot] = i;}
        for(int j=1 ;j<=tot && pri[j]*i<A ;j++){
            vis[i*pri[j]] = 1;
            if(i%pri[j] == 0){
                mu[i*pri[j]] = 0;
                break;
            }
            mu[i*pri[j]] = -mu[i];
        }
    }
    int last;
    for(int i=1 ;i<A ;i++){
        mu[i] += mu[i-1];
        for(int j=1 ;j<=i ;j=last+1){
            last = i/(i/j);
            d[i] += 1LL*(i/j)*(last - j + 1);
        }
    }
}
 
int main(){
    init();
    int T;
    scanf("%d",&T);
    while(T--){
        int n,m,last;
        scanf("%d%d",&n,&m);
        if(n>m) swap(n,m);
        ll ans = 0;
        for(int i=1 ;i<=n ;i=last+1){
            last = min(n/(n/i),m/(m/i));
            ans += 1LL*(mu[last] - mu[i-1])*d[n/i]*d[m/i];
        }
        printf("%lld\n",ans);
    }
    return 0;
}

---

后来看网上博客都说$f(x)$显然是$d(x)$的前缀。
推导了一下确实如此。
对于前n个约数个数之和：
$$\sum _{i=1}^{n}d(i)=\sum _{i=1}^n\sum _{j=1}^n[j|i]=\sum _{i=1}^n\sum _{j=1}^n[i|j]=\sum _{i=1}^n\lfloor \frac{n}{i}\rfloor$$

故可以优化到$O(n)$完成预处理。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
typedef long long ll;
 
const int A = 5e4 + 10;
int pri[A],cnt[A],tot;
bool vis[A];
ll d[A],mu[A];
 
void init(){
    tot = 0;mu[1] = d[1] = 1;
    for(int i=2 ;i<A ;i++){
        if(!vis[i]){mu[i] = -1;pri[++tot] = i;d[i] = 2;cnt[i] = 1;}
        for(int j=1 ;j<=tot && pri[j]*i<A ;j++){
            vis[i*pri[j]] = 1;
            if(i%pri[j] == 0){
                d[i*pri[j]] = d[i]/(cnt[i]+1)*(cnt[i]+2);
                cnt[i*pri[j]] = cnt[i] + 1;
                mu[i*pri[j]] = 0;
                break;
            }
            d[i*pri[j]] = d[i]<<1;
            cnt[i*pri[j]] = 1;
            mu[i*pri[j]] = -mu[i];
        }
    }
    for(int i=1 ;i<A ;i++){
        mu[i] += mu[i-1];
        d[i]  += d[i-1];
    }
}
 
int main(){
    init();
    int T;
    scanf("%d",&T);
    while(T--){
        int n,m,last;
        scanf("%d%d",&n,&m);
        if(n>m) swap(n,m);
        ll ans = 0;
        for(int i=1 ;i<=n ;i=last+1){
            last = min(n/(n/i),m/(m/i));
            ans += 1LL*(mu[last] - mu[i-1])*d[n/i]*d[m/i];
        }
        printf("%lld\n",ans);
    }
    return 0;
}