牛客挑战赛47 F 简单题（莫比乌斯反演）

牛客挑战赛47 F 简单题（莫比乌斯反演）

题解

$当gcd(i,j)>1时：\mu (ij)中包含平方因子gcd(i,j)，此时\mu (ij)=0，因此只需枚举gcd(i,j)=1的情况：$
$$\sum _{i=1}^n\sum _{j=1}^m\phi (ij)\mu (ij)[gcd(i,j)==1]$$$由积性函数的性质可得：当gcd(i,j)=1时：$$\phi (ij)=\phi (i)\phi (j)，\mu (ij)=\mu (i)\mu (j)$$$\sum _{i=1}^n\sum _{j=1}^m\phi (i)\phi (j)\mu (i)\mu (j)[gcd(i,j)==1]$$$$=\sum _{i=1}^n\sum _{j=1}^m\phi (i)\phi (j)\mu (i)\mu (j)\sum _{d|gcd(i,j)}\mu (d)$$$$=\sum _{i=1}^n\phi (i)\mu (i)\sum _{j=1}^m\phi (j)\mu (j)\sum _{d|i,d|j}\mu (d)$$$$=\sum _{d=1}^{min(n,m)}\mu (d)(\sum _{i=1}^n\phi (i)\mu (i)[d|i])(\sum _{j=1}^m\phi (j)\mu (j)[d|j])$$
$令f(d)={\sum }_{i=1}^n\phi (i)\mu (i)[d|i]，g(d)={\sum }_{i=1}^m\phi (i)\mu (i)[d|i]，最后就可以化简成：$$$\sum _{d=1}^{min(n,m)}\mu (d)f(d)g(d)$$
$我们首先筛出函数\phi 和\mu ，此时如果暴力枚举d，然后f(d)和g(d)，时间复杂度为O(nlogn)，会T，但是我们观察到函数f和g的形式就是狄利克雷后缀和的形式，所以可以用狄利克雷后缀和预处理出来函数f和g，再来枚举d，这样可以优化到O(nloglogn)，就可以过了$

代码

#include<iostream>
using namespace std;
using uint = unsigned int;
const int N = 2e7 + 5;
int T, n, m, p[N], phi[N], mu[N], cnt;
bool vis[N]; uint f[N], g[N];
void get(int n) {
    phi[1] = 1, mu[1] = 1;
    for(int i = 2; i <= n; i++) {
        if(!vis[i]) {
            p[cnt++] = i;
            phi[i] = i - 1;
            mu[i] = -1;
        }
        for(int j = 0; p[j] <= n / i; j++) {
            vis[i * p[j]] = 1;
            if(i % p[j] == 0) {
                phi[i * p[j]] = phi[i] * p[j];
                break;
            }
            phi[i * p[j]] = phi[i] * (p[j] - 1);
            mu[i * p[j]] = -mu[i];
        }
    }
}
int main() {
    cin >> T;
    get(N - 5);
    while(T--) {
        cin >> n >> m;
        for(int i = 1; i <= n; i++) f[i] = uint(phi[i] * mu[i]);
        for(int i = 1; i <= m; i++) g[i] = uint(phi[i] * mu[i]);
        for(int i = 0; i < cnt && p[i] <= n; i++)
            for(int j = n / p[i]; j; j--) f[j] += f[j * p[i]];
        for(int i = 0; i < cnt && p[i] <= m; i++)
            for(int j = m / p[i]; j; j--) g[j] += g[j * p[i]];
        uint res = 0;
        for(int i = 1; i <= min(n, m); i++)
            res += mu[i] * f[i] * g[i];
        cout << res << endl;
    }
    return 0;
}