洛谷 P3327 [SDOI2015]约数个数和（莫比乌斯反演）

传送门

---

题目大意

求${\sum }_{i=1}^n{\sum }_{j=1}^{m}d(ij)$，其中$d(ij)$代表$ij$的约数个数

解题思路

一个重要的公式：$d(ij)={\sum }_{x|i}{\sum }_{y|j}[(x,y)=1]$，原式变为：$f(n,m)={\sum }_{i=1}^n{\sum }_{j=1}^m{\sum }_{x|i}{\sum }_{y|j}[(x,y)=1]$

证明：考虑一个质数$p$，在$a$中为$p^{\alpha }$，在$b$中为$p^{\beta }$，在$ij$中对因数个数的贡献为$\alpha +\beta +1$。在等式的右边要满足$gcd(x,y)=1$，就有$gcd(p^{{\alpha }_x},p^{{\beta }_y})=1$，要么${\alpha }_x=0,{\beta }_y\in [0,\beta ]$，有$\beta +1$种；要么${\beta }_y=0,{\alpha }_x\in [0,\alpha ]$，有$\alpha +1$种。最后减去二者同为零的情况，即$\alpha +\beta +1$，等于左式，得证。

根据${\sum }_{d|n}\mu (d)=[\frac{1}{n}]\iff [gcd(x,y)=1]$，得：$f(n,m)={\sum }_{i=1}^n{\sum }_{j=1}^m{\sum }_{x|i}{\sum }_{y|j}{\sum }_{d|(x,y)}\mu (d)$

因为$i,j$会取遍$[1,n],[1,m]$，而$x,y$分别取遍其所有的因子时就可以取遍$[1,n],[1,m]$的所有数，联系到整除分块不难得知$[1,n],[1,m]$的某个因子$x,y$分别有$\lfloor \frac{n}{x}\rfloor ,\lfloor \frac{m}{y}\rfloor$个。那么可以简化为$f(n,m)={\sum }_{x=1}^n{\sum }_{y=1}^m\lfloor \frac{n}{x}\rfloor \lfloor \frac{m}{y}\rfloor {\sum }_{d|(x,y)}\mu (d)$

设$x=x^{\prime }\ast d,y=y^{\prime }\ast d$，然后再次令新的$x=x^{\prime },y=y^{\prime }$，最后得到：

$f(n,m)={\sum }_{d=1}^n\mu (d){\sum }_{x=1}^{\lfloor \frac{n}{d}\rfloor }\lfloor \frac{n}{dx}\rfloor {\sum }_{y=1}^{\lfloor \frac{m}{d}\rfloor }\lfloor \frac{m}{dy}\rfloor$

令$g(n)={\sum }_{i=1}^n\lfloor \frac{n}{i}\rfloor ={\sum }_{i=1}^{n}d(i)$，右式是可以线性筛预处理的，而${\sum }_{d=1}^n\mu (d)$可以求前缀和。那么最终的式子为${\sum }_{d=1}^n\mu (d)g(\lfloor \frac{n}{d}\rfloor )g(\lfloor \frac{m}{d}\rfloor )$，是可以整除分块来求的。

代码

//
// Created by Happig on 2020/9/23
//
#include <bits/stdc++.h>
#include <unordered_map>
#include <unordered_set>

using namespace std;
#define fi first
#define se second
#define pb push_back
#define ins insert
#define Vector Point
#define ENDL "\n"
#define lowbit(x) (x&(-x))
#define mkp(x, y) make_pair(x,y)
#define mem(a, x) memset(a,x,sizeof a);
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef pair<double, double> pdd;
const double eps = 1e-8;
const double pi = acos(-1.0);
const int inf = 0x3f3f3f3f;
const double dinf = 1e300;
const ll INF = 1e18;
const int Mod = 1e9 + 7;
const int maxn = 5e4 + 10;

vector<int> prime;
bitset<maxn> vis;
int mu[maxn], premu[maxn], a[maxn];
ll d[maxn];

void init() {
    mu[1] = d[1] = 1;
    for (int i = 2; i < maxn; i++) {
        if (!vis[i]) {
            prime.push_back(i);
            mu[i] = -1, a[i] = 1, d[i] = 2;
        }
        for (int j = 0; j < prime.size() && i * prime[j] < maxn; j++) {
            vis[i * prime[j]] = 1;
            if (i % prime[j]) {
                mu[i * prime[j]] = -mu[i];
                d[i * prime[j]] = d[i] * 2, a[i * prime[j]] = 1;
            } else {
                mu[i * prime[j]] = 0;
                d[i * prime[j]] = d[i] / (a[i] + 1) * (a[i] + 2), a[i * prime[j]] = a[i] + 1;
                break;
            }
        }
    }
    for (int i = 1; i < maxn; i++) {
        premu[i] = premu[i - 1] + mu[i];
        d[i] += d[i - 1];
    }
}


int main() {
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    ios_base::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int t, n, m;
    cin >> t;
    init();
    while (t--) {
        cin >> n >> m;
        int up = min(n, m);
        ll ans = 0;
        for (int l = 1, r; l <= up; l = r + 1) {
            r = min(n / (n / l), m / (m / l));
            if (r > up) r = up;
            ans += (premu[r] - premu[l - 1]) * d[n / l] * d[m / l];
        }
        cout << ans << ENDL;
    }
    return 0;
}