bzoj 2820: YY的GCD （反演）

题目描述

传送门

题目大意：
$\sum\limits_{p}^{min(n,m)}\sum\limits_{i=1}^n\sum\limits_{j=1}^m [gcd(i,j)=p]$
其中p是质数。

题解

这道题还是要用到反演啊
如果$F(n)=\sum\limits_{n|d} f(d)$,那么$f(n)=\sum\limits_{n|d}\mu({d\over n})*F(d)$
设$f(d)= \sum_{i=1}^n\sum_{j=1}^m [gcd(i,j)=d]$ 表示满足$gcd(i,j)=d$且$1<=i<=n,1<=j<=m$的$(i,j)$的对数
$F(d)=\sum_{d|n} f(n)$ 表示满足$d|gcd(i,j)$且$1<=i<=n,1<=j<=m$的$(i,j)$的对数
$F(d)={\lfloor {n\over d}\rfloor}*{\lfloor {m \over d}\rfloor }$
设n<=m,那么上面的式子就转化成了

$$\sum\limits_{p}^{min(n,m)}\sum\limits_{p|d}\mu({d\over p})*F(d)$$
$$\sum\limits_{p}^{min(n,m)}\sum\limits_{d=1}^n \mu(d)*F(d*p)$$
然后令T=d*p，改变枚举的顺序
$$\sum\limits_{T=1}^n {\lfloor {n\over T}\rfloor}*{\lfloor {m \over T}\rfloor } \sum\limits_{p|T} \mu({T\over p})$$ 其中p是质数
那么如果我们能预处理 $\sum\limits_{p|T} \mu({T\over p})$，这个问题就能在 $O(\sqrt n+\sqrt m)$
我们可以直接暴力枚举质数p，然后用p去更新所有范围内p的倍数。
貌似n内质数的个数是接近 $n/\log n$,对于每个素数来说，枚举倍数的均摊复杂度是 $O(logn)$,所以暴力预处理的复杂度是接近O(n)的。

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define N 10000000
#define LL long long 
using namespace std;
int prime[N+3],mu[N+3];
LL f[N+3];
bool pd[N+3];
int n,m,T;
void init()
{
    mu[1]=1;
    for (int i=2;i<=N;i++) {
        if (!pd[i]) {
            prime[++prime[0]]=i;
            mu[i]=-1;
        }
        for (int j=1;j<=prime[0];j++) {
            if (i*prime[j]>N) break;
            pd[i*prime[j]]=1;
            if (i%prime[j]==0) {
                mu[i*prime[j]]=0;
                break;
            }
            mu[i*prime[j]]=-mu[i];
        }
    }
    for (int i=1;i<=prime[0];i++) {
        int t=prime[i];
        for (int j=1;j*t<=N;j++) f[j*t]+=mu[j];
    }
    for (int i=1;i<=N;i++) f[i]=f[i-1]+f[i];
}
int main()
{
    freopen("a.in","r",stdin);
    freopen("my.out","w",stdout);
    scanf("%d",&T);
    init();
    while (T--) {
        scanf("%d%d",&n,&m);
        if (n>m) swap(n,m);
        int j=0; LL ans=0;
        for (int i=1;i<=n;i=j+1) {
            j=min(n/(n/i),m/(m/i));
            ans=ans+(f[j]-f[i-1])*(n/i)*(m/i);
        }
        printf("%lld\n",ans);
    } 
}